Removed unecessary imports [ci skip]

miceforest/ImputationSchema.py

@@ -1,6 +1,6 @@
 from numpy import concatenate
 from .utils import _copy_and_remove, _setequal, _list_union, MeanMatchType, VarSchemType
-from typing import Optional, Union, List, Dict, TYPE_CHECKING, Callable
+from typing import Union, List, TYPE_CHECKING, Callable
 from pandas import unique
 
 if TYPE_CHECKING:

miceforest/ImputedDataSet.py

@@ -1,7 +1,7 @@
 from .ImputationSchema import _ImputationSchema
 import numpy as np
 from pandas import DataFrame
-from .utils import ensure_rng, _list_union, _var_comparison, MeanMatchType, VarSchemType
+from .utils import ensure_rng, _var_comparison, MeanMatchType, VarSchemType
 from typing import Optional, Union, List, Dict, Callable
 
 

miceforest/KernelDataSet.py

@@ -10,13 +10,10 @@
 )
 from pandas import DataFrame
 import numpy as np
-from typing import Union, List, Dict, Any, TYPE_CHECKING, Callable
+from typing import Union, Dict, Any, Callable
 from .logger import Logger
 from lightgbm import train, Dataset
 
-# if TYPE_CHECKING:
-#     from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
-
 _TIMED_EVENTS = ["mice", "model_fit", "model_predict", "mean_match", "impute_new_data"]
 
 

miceforest/MultipleImputedDataSet.py

@@ -37,8 +37,6 @@ def __init__(self, initial_dataset: ImputedDataSet):
         )
         self.data = getattr(initial_dataset, "data")
         self.save_all_iterations = getattr(initial_dataset, "save_all_iterations")
-        # self.categorical_variables = getattr(initial_dataset, "categorical_variables")
-        # self._varfilter = getattr(initial_dataset, "_varfilter")
         self._prep_multi_plot = getattr(initial_dataset, "_prep_multi_plot")
         self._default_iteration = getattr(initial_dataset, "_default_iteration")
         self.imputed_data_sets = {0: initial_dataset}

miceforest/MultipleImputedKernel.py

@@ -2,7 +2,7 @@
 from .MultipleImputedDataSet import MultipleImputedDataSet
 from .KernelDataSet import KernelDataSet
 from pandas import DataFrame
-from typing import Union, List, Dict, Callable
+from typing import Union, List, Callable
 from .utils import ensure_rng, MeanMatchType, VarSchemType
 from .logger import Logger