diff --git a/src/anemoi/inference/metadata.py b/src/anemoi/inference/metadata.py index e5dc0ab..5eb674a 100644 --- a/src/anemoi/inference/metadata.py +++ b/src/anemoi/inference/metadata.py @@ -512,11 +512,12 @@ def _find(x): _find(y) if isinstance(x, dict): - if "dataset" in x and isinstance(x["dataset"], str): + if "dataset" in x and isinstance(x["dataset"], str): result.append(x["dataset"]) for k, v in x.items(): _find(v) + _find(self._config.dataloader.training.dataset) return result