Apply automatic code formatting

Author: actions-user

lazypredict/models/classification.py

@@ -10,12 +10,7 @@
 import numpy as np
 import pandas as pd
 from sklearn.base import ClassifierMixin
-from sklearn.metrics import (
-    accuracy_score,
-    balanced_accuracy_score,
-    f1_score,
-    roc_auc_score,
-)
+from sklearn.metrics import accuracy_score, balanced_accuracy_score, f1_score, roc_auc_score
 from sklearn.pipeline import Pipeline
 from tqdm import tqdm
 

lazypredict/utils/__init__.py

@@ -6,21 +6,9 @@
 import numpy as np
 import pandas as pd
 
-from .base import (
-    BaseLazy,
-    check_X_y,
-    get_model_name,
-)
-from .gpu import (
-    get_best_model,
-    get_cpu_model,
-    is_cuml_available,
-    is_gpu_available,
-)
-from .metrics import (
-    get_classification_metrics,
-    get_regression_metrics,
-)
+from .base import BaseLazy, check_X_y, get_model_name
+from .gpu import get_best_model, get_cpu_model, is_cuml_available, is_gpu_available
+from .metrics import get_classification_metrics, get_regression_metrics
 from .mlflow_utils import (
     configure_mlflow,
     end_run,
@@ -32,11 +20,7 @@
     log_params,
     start_run,
 )
-from .preprocessing import (
-    categorical_cardinality_threshold,
-    create_preprocessor,
-    get_card_split,
-)
+from .preprocessing import categorical_cardinality_threshold, create_preprocessor, get_card_split
 
 logger = logging.getLogger("lazypredict.utils")
 

lazypredict/utils/preprocessing.py

@@ -10,12 +10,7 @@
 from sklearn.compose import ColumnTransformer
 from sklearn.impute import SimpleImputer
 from sklearn.pipeline import Pipeline
-from sklearn.preprocessing import (
-    OneHotEncoder,
-    OrdinalEncoder,
-    PolynomialFeatures,
-    StandardScaler,
-)
+from sklearn.preprocessing import OneHotEncoder, OrdinalEncoder, PolynomialFeatures, StandardScaler
 
 logger = logging.getLogger("lazypredict.preprocessing")
 

tests/test_base.py

@@ -155,10 +155,7 @@ def setUp(self):
 
     def test_metrics_import(self):
         try:
-            from lazypredict.metrics import (
-                get_classification_metrics,
-                get_regression_metrics,
-            )
+            from lazypredict.metrics import get_classification_metrics, get_regression_metrics
 
             self.assertTrue(callable(get_classification_metrics))
             self.assertTrue(callable(get_regression_metrics))

tests/test_helpers.py

@@ -11,9 +11,7 @@ class TestHelpers(unittest.TestCase):
 
     def test_get_card_split(self):
         """Test categorical cardinality threshold function."""
-        from lazypredict.utils.preprocessing import (
-            categorical_cardinality_threshold,
-        )
+        from lazypredict.utils.preprocessing import categorical_cardinality_threshold
 
         # Create test data with known cardinality
         data = pd.DataFrame(

tests/test_integration.py

@@ -110,10 +110,7 @@ def test_regression_direct_import(self):
     def test_supervised_backwards_compatibility(self):
         try:
             # Test with minimal models (just one) to keep tests fast
-            from sklearn.tree import (
-                DecisionTreeClassifier,
-                DecisionTreeRegressor,
-            )
+            from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
 
             from lazypredict.Supervised import LazyClassifier, LazyRegressor
 

tests/test_model_registry.py

@@ -12,10 +12,7 @@ class TestModelRegistry(unittest.TestCase):
     def test_classification_registry(self):
         """Test classification model registry."""
         try:
-            from lazypredict.models.model_registry import (
-                CLASSIFIERS,
-                get_classification_models,
-            )
+            from lazypredict.models.model_registry import CLASSIFIERS, get_classification_models
 
             # Test getting all models
             models = get_classification_models()
@@ -37,10 +34,7 @@ def test_classification_registry(self):
     def test_regression_registry(self):
         """Test regression model registry."""
         try:
-            from lazypredict.models.model_registry import (
-                REGRESSORS,
-                get_regression_models,
-            )
+            from lazypredict.models.model_registry import REGRESSORS, get_regression_models
 
             # Test getting all models
             models = get_regression_models()
@@ -62,10 +56,7 @@ def test_regression_registry(self):
     def test_classifier_filter(self):
         """Test classifier filtering."""
         try:
-            from lazypredict.models.model_registry import (
-                CLASSIFIERS,
-                filter_models,
-            )
+            from lazypredict.models.model_registry import CLASSIFIERS, filter_models
 
             # Test filtering with exclude list
             filtered = filter_models(CLASSIFIERS, exclude=["RandomForestClassifier"])
@@ -85,10 +76,7 @@ def test_classifier_filter(self):
     def test_regressor_filter(self):
         """Test regressor filtering."""
         try:
-            from lazypredict.models.model_registry import (
-                REGRESSORS,
-                filter_models,
-            )
+            from lazypredict.models.model_registry import REGRESSORS, filter_models
 
             # Test filtering with exclude list
             filtered = filter_models(REGRESSORS, exclude=["RandomForestRegressor"])
@@ -108,9 +96,7 @@ def test_regressor_filter(self):
     def test_classifier_initialization(self):
         """Test classifier initialization."""
         try:
-            from lazypredict.models.model_registry import (
-                get_classification_models,
-            )
+            from lazypredict.models.model_registry import get_classification_models
 
             models = get_classification_models()
             for model_class in models:

tests/test_supervised.py

@@ -158,11 +158,7 @@ def test_mlflow_integration(self):
 
 import unittest
 
-from lazypredict import (
-    LazyClassifier,
-    LazyRegressor,
-    Supervised,
-)
+from lazypredict import LazyClassifier, LazyRegressor, Supervised
 
 
 class TestSupervisedModule(unittest.TestCase):

tests/test_utils.py

@@ -101,9 +101,7 @@ def setUp(self):
     def test_categorical_cardinality_threshold(self):
         """Test categorical cardinality threshold function."""
         try:
-            from lazypredict.utils.preprocessing import (
-                categorical_cardinality_threshold,
-            )
+            from lazypredict.utils.preprocessing import categorical_cardinality_threshold
 
             data = pd.DataFrame(
                 {