Let ManualScaler inherit from the corresponding protocol

opcode81 · Feb 21, 2024 · c439c34 · c439c34
1 parent ab40f2b
commit c439c34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sensai/data_transformation/sklearn_transformer.py b/src/sensai/data_transformation/sklearn_transformer.py
@@ -30,7 +30,7 @@ def fit(self, arr: TransformableArray):
         pass
 
 
-class ManualScaler:
+class ManualScaler(SkLearnTransformerProtocol):
     """
     A scaler whose parameters are not learnt from data but manually defined
     """