Merge pull request #299 from abstractqqq/sklearn_ver_in_test

abstractqqq · web-flow · commit 8b0640b93cef · 2024-12-10T00:58:36.000-05:00
diff --git a/tests/requirements-test.txt b/tests/requirements-test.txt
@@ -3,7 +3,7 @@ maturin[patchelf]>=1.7; sys_platform == "linux"
 maturin>=1.7; sys_platform != "linux"
 polars
 numpy
-scikit-learn
+scikit-learn<=1.5
 scipy
 pyarrow
 pandas
diff --git a/tests/test_transforms.py b/tests/test_transforms.py
@@ -163,7 +163,6 @@ def test_target_encode():
     enc.fit(df_pd[cols], y)
     df_transformed = enc.transform(df_pd[cols])
     df2 = pl.from_pandas(df_transformed[cols])  # output from category_encoders
-
     assert_frame_equal(df1, df2)
 
 
