Merge remote-tracking branch 'origin/main' into progress_tracker

Author: sam-laing

oellm_autoexp/monitor/local_client.py

@@ -17,7 +17,7 @@
 import yaml
 import logging
 
-from compoconf import ConfigInterface, register
+from compoconf import ConfigInterface, register, asdict
 
 from oellm_autoexp.monitor.job_client_protocol import JobClientInterface
 from oellm_autoexp.monitor.utils.paths import resolve_log_path, update_log_symlink
@@ -104,7 +104,7 @@ def submit(
                 LOGGER.info(f"Logging Config to: {resolved_config_path}")
                 Path(resolved_config_path).parent.mkdir(parents=True, exist_ok=True)
                 with open(resolved_config_path, "w") as fp:
-                    yaml.dump(job.base_config, fp)
+                    yaml.dump(asdict(job.base_config), fp)
                 if job.config_path_current:
                     config_path_obj = Path(resolved_config_path)
                     update_log_symlink(config_path_obj, Path(job.config_path_current))

oellm_autoexp/monitor/slurm_client.py

@@ -7,7 +7,7 @@
 from pathlib import Path
 import yaml
 
-from compoconf import ConfigInterface, register
+from compoconf import ConfigInterface, register, asdict
 from oellm_autoexp.slurm_gen import generate_script, validate_job_script
 from oellm_autoexp.slurm_gen.client import (
     BaseSlurmClient,
@@ -83,7 +83,7 @@ def submit(
             config_path = expand_log_path(job.config_path, job_id=job_id)
             LOGGER.info(f"Logging Config to: {config_path}")
             with open(config_path, "w") as fp:
-                yaml.dump(job.base_config, fp)
+                yaml.dump(asdict(job.base_config), fp)
             if job.config_path_current:
                 update_log_symlink(Path(config_path), Path(job.config_path_current))
         return job_id