diff --git a/align_system/cli/run_align_system.py b/align_system/cli/run_align_system.py index 382f8ac..4cccd02 100644 --- 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: