Code cleanup (#234)

scripts/run_eval.py

@@ -52,7 +52,7 @@ def _load_checkpoint(config: DictConfig, graph: Graph) -> None:
     if config.load_step is None:
         print("Loading latest checkpoint from load_dir")
         # NOTE: this is specific to the checkpoint name format
-        load_step = sorted([int(x[x.find("-") + 1 : x.find(".")]) for x in os.listdir(config.load_dir)])[-1]
+        load_step = sorted(int(x[x.find("-") + 1 : x.find(".")]) for x in os.listdir(config.load_dir))[-1]
     else:
         load_step = config.load_step
     load_path = os.path.join(config.load_dir, f"step-{load_step:09d}.ckpt")

scripts/run_train.py

@@ -212,7 +212,7 @@ def main(config: DictConfig):
 
     unrolled_path = os.path.join(os.getcwd(), ".hydra/config.yaml")
     if os.path.exists(unrolled_path):
-        with open(unrolled_path, "r", encoding="utf8") as f:
+        with open(unrolled_path, encoding="utf8") as f:
             unrolled_config = yaml.safe_load(f)
         logger = logging.getLogger(__name__)
         logger.info("Printing current config setup")