align new imputer with others

ecmwf · Sep 13, 2024 · 829c6f7 · 829c6f7
1 parent 6c2d8d1
commit 829c6f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/anemoi/models/preprocessing/imputer.py b/src/anemoi/models/preprocessing/imputer.py
@@ -207,7 +207,7 @@ class ConstantAdditionalNaNImputer(BaseImputer):
     def __init__(
         self,
         config=None,
-        data_indices: Optional[dict] = None,
+        data_indices: Optional[IndexCollection] = None,
         statistics: Optional[dict] = None,
     ) -> None:
         super().__init__(config, data_indices, statistics)