fix default config replaced agent

Author: kew6688

scripts/eval/eval_iros.py

@@ -1,13 +1,16 @@
 import sys
+
 sys.path.append('.')
 
-from internnav.configs.evaluator.default_config import get_config
-from internnav.evaluator import Evaluator
 import argparse
 import importlib.util
 
+from internnav.configs.evaluator.default_config import get_config
+from internnav.evaluator import Evaluator
+
 # This file is the main file
 
+
 def parse_args():
     parser = argparse.ArgumentParser()
     parser.add_argument(
@@ -32,19 +35,21 @@ def parse_args():
     )
     return parser.parse_args()
 
+
 def load_eval_cfg(config_path, attr_name='eval_cfg'):
     spec = importlib.util.spec_from_file_location("eval_config_module", config_path)
     config_module = importlib.util.module_from_spec(spec)
     sys.modules["eval_config_module"] = config_module
     spec.loader.exec_module(config_module)
     return getattr(config_module, attr_name)
 
+
 def replace_cfg(evaluator_cfg, default_cfg, split):
-    default_cfg.agent.model_name = evaluator_cfg.agent.model_name
-    default_cfg.agent.ckpt_path = evaluator_cfg.agent.ckpt_path
+    default_cfg.agent = evaluator_cfg.agent
     if split:
         default_cfg.dataset.dataset_settings['split_data_types'] = [split]
 
+
 def main():
     args = parse_args()
     evaluator_cfg = load_eval_cfg(args.config, attr_name='eval_cfg')
@@ -57,5 +62,6 @@ def main():
     evaluator.save_to_json = True
     evaluator.eval()
 
+
 if __name__ == '__main__':
-    main()
+    main()