update the method for adding values to the arguments in the config file

openproblems-bio · May 4, 2024 · 124a7cf · 124a7cf
1 parent c9044f4
commit 124a7cf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/common/create_component/script.py b/src/common/create_component/script.py
@@ -170,11 +170,10 @@ def set_par_values(config) -> None:
       value = arg.get("default", arg.get("example", "..."))
     elif key == "de_train":
       value = "resources/neurips-2023-kaggle/de_train.parquet"
+    elif key == "de_train":
+      value = "resources/neurips-2023-kaggle/de_train.h5ad"
     elif key == "id_map":
       value = "resources/neurips-2023-kaggle/id_map.csv"
-    elif arg.get("direction", "input") == "input":
-      key_strip = key.replace("input_", "")
-      value = f'resources_test/{par["task"]}/pancreas/{key_strip}.h5ad'
     else:
       key_strip = key.replace("output_", "")
       value = f'{key_strip}.h5ad'