fix: apply filter to inherited datasets

radiocosmology · May 3, 2024 · 7b340e3 · 7b340e3
1 parent 3a059dc
commit 7b340e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/draco/core/containers.py b/draco/core/containers.py
@@ -480,7 +480,7 @@ def _make_selections(cls, sel_args):
 
         # Build selections dict
         selections = {}
-        for name, dataset in cls._dataset_spec.items():
+        for name, dataset in cls._class_dataset_spec().items():
             ds_axes = dataset["axes"]
             sel = []
             ds_relevant = False