fix sklearn version check formatting

JacksonBurns · Jul 1, 2023 · 9de8085 · 9de8085
1 parent a06c0a5
commit 9de8085
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 16 deletions.
diff --git a/test/regression/test_regression.py b/test/regression/test_regression.py
@@ -14,14 +14,7 @@
 )
 
 SKLEARN_GEQ_13 = (  # get the sklearn version
-    int(
-        pkg_resources.get_distribution(
-            "scikit-learn",
-        ).version.split(
-            "."
-        )[1]
-    )
-    >= 3
+    int(pkg_resources.get_distribution("scikit-learn").version.split(".")[1]) >= 3
 )
 
 

diff --git a/test/unit/samplers/extrapolative/test_kmeans.py b/test/unit/samplers/extrapolative/test_kmeans.py
@@ -8,14 +8,7 @@
 from astartes.utils.warnings import ImperfectSplittingWarning
 
 SKLEARN_GEQ_13 = (  # get the sklearn version
-    int(
-        pkg_resources.get_distribution(
-            "scikit-learn",
-        ).version.split(
-            "."
-        )[1]
-    )
-    >= 3
+    int(pkg_resources.get_distribution("scikit-learn").version.split(".")[1]) >= 3
 )