feat: check_estimator imputers.Imputer()

Author: anhkhoangoho

qolmat/imputations/imputers.py

@@ -1,5 +1,6 @@
 import warnings
 from typing import Dict, List, Optional, Union
+from abc import abstractmethod
 
 import numpy as np
 import pandas as pd
@@ -187,6 +188,10 @@ def impute_element(self, df: pd.DataFrame) -> pd.DataFrame:
 
         return df
 
+    @abstractmethod
+    def fit_transform_element(self, df: pd.DataFrame):
+        return df
+
 
 class ImputerOracle(Imputer):
     """

tests/imputations/test_imputers.py

@@ -234,6 +234,7 @@ def test_ImputerEM_fit_transform(df: pd.DataFrame) -> None:
 
 @parametrize_with_checks(
     [
+        imputers.Imputer(),
         imputers.ImputerOracle(),
         imputers.ImputerMean(),
         imputers.ImputerMedian(),