Unverified Commit aa2dd2b5 authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

[Format] Add config lints (#892)

parent 3dbba119
......@@ -20,7 +20,7 @@ for _name in subset_mapping.keys():
llm_cmp_infer_cfg = dict(
prompt_template=dict(
type=PromptTemplate,
template="{content}",
template='{content}',
),
# No in-context example, using ZeroRetriever
retriever=dict(type=ZeroRetriever),
......@@ -34,13 +34,13 @@ for _name in subset_mapping.keys():
llm_compression_datasets.append(
dict(
abbr=f"llm_compression-{_name}",
abbr=f'llm_compression-{_name}',
type=LLMCompressionDataset,
path="./data/llm-compression",
path='./data/llm-compression',
name=_name,
samples=None, # Set small samples for testing
reader_cfg=dict(
input_columns=["content"],
input_columns=['content'],
output_column=None,
),
infer_cfg=llm_cmp_infer_cfg,
......
......@@ -34,5 +34,5 @@ with read_base():
)
LVEval_datasets = sum(
(v for k, v in locals().items() if k.endswith("_datasets")), []
(v for k, v in locals().items() if k.endswith('_datasets')), []
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment