provide polars suport with new scikit-learn 1.4.0 API

paucablop · Feb 11, 2024 · cb9056e · cb9056e
1 parent 3169661
commit cb9056e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -27,8 +27,10 @@
     install_requires=[
         "numpy",
         "pandas",
+        "polars",
+        "pyarrow",
         "scipy",
-        "scikit-learn",
+        "scikit-learn>=1.4.0",
     ],
     include_package_data=True,
     package_data={'': ['tests/resources/*.csv',