Fix output dirs

BioinfoMachineLearning · Dec 8, 2024 · 952120e · 952120e
1 parent 115dda4
commit 952120e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/posebench/models/diffdock_inference.py b/posebench/models/diffdock_inference.py
@@ -60,7 +60,7 @@ def main(cfg: DictConfig):
 
         if cfg.max_num_inputs:
             cfg.output_dir = cfg.output_dir.replace(
-                f"_{cfg.dataset}", f"_first_{cfg.max_num_inputs}_{cfg.dataset}"
+                f"_{cfg.dataset}", f"_{cfg.dataset}_first_{cfg.max_num_inputs}"
             )
 
     assert os.path.exists(input_csv_path), f"Input CSV file `{input_csv_path}` not found."

diff --git a/posebench/models/flowdock_inference.py b/posebench/models/flowdock_inference.py
@@ -43,15 +43,15 @@ def main(cfg: DictConfig):
         if cfg.pocket_only_baseline:
             cfg.out_path = os.path.join(
                 os.path.dirname(cfg.out_path),
-                os.path.basename(cfg.out_path).replace("neuralplexer", "neuralplexer_pocket_only"),
+                os.path.basename(cfg.out_path).replace("flowdock", "flowdock_pocket_only"),
             )
-            input_csv_path = cfg.input_csv_path.replace("neuralplexer", "neuralplexer_pocket_only")
+            input_csv_path = cfg.input_csv_path.replace("flowdock", "flowdock_pocket_only")
 
         if cfg.max_num_inputs:
             cfg.out_path = os.path.join(
                 os.path.dirname(cfg.out_path),
                 os.path.basename(cfg.out_path).replace(
-                    "neuralplexer", f"neuralplexer_first_{cfg.max_num_inputs}"
+                    "flowdock", f"flowdock_first_{cfg.max_num_inputs}"
                 ),
             )