mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-25 17:10:42 +00:00
Update optimizer_benchmark_env.py
This commit is contained in:
parent
291dcd8351
commit
f9a444b6f2
1 changed files with 2 additions and 2 deletions
|
|
@ -47,7 +47,7 @@ class OptimizerBenchmarkEnv:
|
|||
if self.optimizer_name is None:
|
||||
raise ValueError("No optimizer registered. Call submit_optimizer first.")
|
||||
if output_dir is None:
|
||||
output_dir = f"outputs/experiment_{self.optimizer_name}"
|
||||
output_dir = f"FOB/outputs/experiment_{self.optimizer_name}"
|
||||
if yaml_path is None:
|
||||
yaml_path = self.fob_root / f"experiment_{self.optimizer_name}.yaml"
|
||||
else:
|
||||
|
|
@ -206,4 +206,4 @@ if __name__ == "__main__":
|
|||
test_optimizer_registration()
|
||||
test_yaml_generation()
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue