Add ListConfig as a possible instance type

ashleve · Oct 14, 2023 · f37ebdf · f37ebdf
1 parent bddbc24
commit f37ebdf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils/rich_utils.py b/src/utils/rich_utils.py
@@ -6,7 +6,7 @@
 import rich.tree
 from hydra.core.hydra_config import HydraConfig
 from lightning_utilities.core.rank_zero import rank_zero_only
-from omegaconf import DictConfig, OmegaConf, open_dict
+from omegaconf import ListConfig, DictConfig, OmegaConf, open_dict
 from rich.prompt import Prompt
 
 from src.utils import pylogger
@@ -58,7 +58,7 @@ def print_config_tree(
         branch = tree.add(field, style=style, guide_style=style)
 
         config_group = cfg[field]
-        if isinstance(config_group, DictConfig):
+        if isinstance(config_group, DictConfig | ListConfig):
             branch_content = OmegaConf.to_yaml(config_group, resolve=resolve)
         else:
             branch_content = str(config_group)