Fix inference outputs dir for Chai-1 pocket only exps

amorehead · amorehead · commit 9aa9453cbcc7 · 2024-09-26T10:23:21.000-05:00
diff --git a/posebench/data/chai_output_extraction.py b/posebench/data/chai_output_extraction.py
@@ -71,7 +71,7 @@ def main(cfg: DictConfig):
                 cfg.dataset, f"{cfg.dataset}_pocket_only"
             )
             cfg.inference_outputs_dir = cfg.inference_outputs_dir.replace(
-                f"rfaa_{cfg.dataset}", f"rfaa_pocket_only_{cfg.dataset}"
+                f"chai-lab_{cfg.dataset}", f"chai-lab_pocket_only_{cfg.dataset}"
             )
 
     if cfg.complex_filepath is not None:

Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,7 @@ def main(cfg: DictConfig):`
`71`	`71`	`cfg.dataset, f"{cfg.dataset}_pocket_only"`
`72`	`72`	`)`
`73`	`73`	`cfg.inference_outputs_dir = cfg.inference_outputs_dir.replace(`
`74`		`- f"rfaa_{cfg.dataset}", f"rfaa_pocket_only_{cfg.dataset}"`
	`74`	`+ f"chai-lab_{cfg.dataset}", f"chai-lab_pocket_only_{cfg.dataset}"`
`75`	`75`	`)`
`76`	`76`
`77`	`77`	`if cfg.complex_filepath is not None:`