Fix save config.py to config.yaml (#322)

medipixel · Dec 13, 2021 · c5f7d1d · c5f7d1d
1 parent 96bceb9
commit c5f7d1d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rl_algorithms/common/abstract/distributed_logger.py b/rl_algorithms/common/abstract/distributed_logger.py
@@ -129,7 +129,7 @@ def set_wandb(self):
             max_episode_steps=self.max_episode_steps,
         )
         wandb.config.update(additional_log)
-        shutil.copy(self.log_cfg.cfg_path, os.path.join(wandb.run.dir, "config.py"))
+        shutil.copy(self.log_cfg.cfg_path, os.path.join(wandb.run.dir, "config.yaml"))
 
     def recv_log_info(self):
         """Receive info from learner."""

diff --git a/rl_algorithms/common/abstract/learner.py b/rl_algorithms/common/abstract/learner.py
@@ -70,7 +70,7 @@ def __init__(
             os.makedirs(self.ckpt_path, exist_ok=True)
 
             # save configuration
-            shutil.copy(log_cfg.cfg_path, os.path.join(self.ckpt_path, "config.py"))
+            shutil.copy(log_cfg.cfg_path, os.path.join(self.ckpt_path, "config.yaml"))
 
     @abstractmethod
     def _init_network(self):