diff --git a/julearn/transformers/dataframe/filter_columns.py b/julearn/transformers/dataframe/filter_columns.py index cf1e16cbe..84ca07899 100644 --- a/julearn/transformers/dataframe/filter_columns.py +++ b/julearn/transformers/dataframe/filter_columns.py @@ -55,8 +55,8 @@ def __init__( def _fit( self, - X: pd.DataFrame, - y: Optional[DataLike] = None, # noqa: N803 + X: pd.DataFrame, # noqa: N803 + y: Optional[DataLike] = None, ) -> "FilterColumns": """Fit the transformer.