From 829c6f76683d659f6def0493a8b4606f95766b14 Mon Sep 17 00:00:00 2001 From: sahahner Date: Fri, 13 Sep 2024 08:21:30 +0000 Subject: [PATCH] align new imputer with others --- src/anemoi/models/preprocessing/imputer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/anemoi/models/preprocessing/imputer.py b/src/anemoi/models/preprocessing/imputer.py index b84a2d8..6251067 100644 --- 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)