Linting

Author: SamoraHunter

ml_grid/model_classes/h2o_rulefit_classifier_class.py

@@ -6,7 +6,7 @@
 """
 
 import logging
-from typing import Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
 import pandas as pd  # type: ignore
 from skopt.space import Categorical, Integer

ml_grid/model_classes/keras_classifier_class.py

@@ -1,12 +1,11 @@
-import math
-
 """Keras Classifier.
 
 This module contains the kerasClassifier_class, which is a configuration
 class for a Keras Sequential model wrapped by KerasClassifier. It provides
 parameter spaces for grid search and Bayesian optimization.
 """
 
+import math
 from typing import Any, Dict, Optional
 
 import numpy as np

ml_grid/model_classes/light_gbm_class.py

@@ -6,7 +6,7 @@ class for the LightGBMClassifier. It provides parameter spaces for
 """
 
 import logging
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
 import pandas as pd
 from skopt.space import Categorical, Integer, Real

tests/test_project_score_save.py

@@ -14,7 +14,7 @@
 if project_root not in sys.path:
     sys.path.insert(0, project_root)
 
-from ml_grid.util.project_score_save import project_score_save_class
+from ml_grid.util.project_score_save import project_score_save_class  # noqa: E402
 
 
 class TestProjectScoreSave(unittest.TestCase):

tests/test_tabpfn_classifier_class.py

@@ -33,7 +33,9 @@
 sys.modules["tabpfn"] = mock_tabpfn_module
 sys.modules["tabpfn.constants"] = MagicMock()
 
-from ml_grid.model_classes.tabpfn_classifier_class import TabPFNClassifierClass
+from ml_grid.model_classes.tabpfn_classifier_class import (
+    TabPFNClassifierClass,
+)  # noqa: E402
 
 
 class TestTabPFNClassifierClass(unittest.TestCase):