Quickfix for alignment target conversion when loading from hydra config

ITM-Kitware · Oct 16, 2024 · e52e4fc · e52e4fc
1 parent 95f05b5
commit e52e4fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/align_system/cli/run_align_system.py b/align_system/cli/run_align_system.py
@@ -137,8 +137,9 @@ def _compute_time_stats(times_s):
             # some nested conversion to dict (from OmegaConf objects)
             # otherwise this can cause some downstream issues with
             # serialization
-            alignment_target.kdma_values = [OmegaConf.to_container(c) for c in
-                                            alignment_target.kdma_values]
+            alignment_target.kdma_values = [OmegaConf.to_container(c)
+                                            if isinstance(c, DictConfig) else c
+                                            for c in alignment_target.kdma_values]
         elif cfg.align_to_target:
             alignment_target = scenario.get_alignment_target()
         else: