Alina Lozovskaya commited on
Commit
1e75b2e
·
1 Parent(s): 3d76e98

Fix config generation

Browse files
Files changed (1) hide show
  1. yourbench_space/config.py +3 -2
yourbench_space/config.py CHANGED
@@ -18,7 +18,7 @@ def generate_base_config(
18
  "hf_organization": hf_org,
19
  "hf_dataset_name": hf_dataset_name,
20
  },
21
- "local_dataset_dir": "results/",
22
  "model_list": [
23
  {
24
  "model_name": model_name,
@@ -44,6 +44,7 @@ def generate_base_config(
44
  },
45
  "upload_ingest_to_hub": {
46
  "source_documents_dir": "/app/example/ingested",
 
47
  "run": True,
48
  },
49
  "summarization": {"run": True},
@@ -83,4 +84,4 @@ def generate_base_config(
83
  def save_config(yaml_text):
84
  with open(CONFIG_PATH, "w") as file:
85
  file.write(yaml_text)
86
- return "✅ Config saved!"
 
18
  "hf_organization": hf_org,
19
  "hf_dataset_name": hf_dataset_name,
20
  },
21
+ "local_dataset_dir": "/app/results/",
22
  "model_list": [
23
  {
24
  "model_name": model_name,
 
44
  },
45
  "upload_ingest_to_hub": {
46
  "source_documents_dir": "/app/example/ingested",
47
+ "hub_dataset_name": f"{hf_dataset_name}_ingested_documents",
48
  "run": True,
49
  },
50
  "summarization": {"run": True},
 
84
  def save_config(yaml_text):
85
  with open(CONFIG_PATH, "w") as file:
86
  file.write(yaml_text)
87
+ return "✅ Config saved!"