pin versions to later versions, replace one_hot_encoder feature name …

…getter
numerai · Dec 20, 2023 · ad8e288 · ad8e288
1 parent deedee4
commit ad8e288
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/numerai_tools/scoring.py b/numerai_tools/scoring.py
@@ -308,7 +308,7 @@ def one_hot_encode(
         one_hot = encoder.fit_transform(df[[col]])
         one_hot = pd.DataFrame(
             one_hot.toarray(),
-            columns=encoder.get_feature_names(),
+            columns=encoder.get_feature_names_out(),
             index=df.index,
         )
         df = df.join(one_hot).drop(columns=col)

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 from setuptools import find_packages
 
 
-VERSION = "0.0.15"
+VERSION = "0.0.16"
 
 
 def load(path):
@@ -37,9 +37,9 @@ def load(path):
         package_data={"numerai": ["LICENSE", "README.md"]},
         packages=find_packages(exclude=["tests"]),
         install_requires=[
-            "pandas",
-            "numpy",
-            "scipy",
-            "scikit-learn",
+            "pandas>=1.3.1, <=2.1.3",
+            "numpy~=1.26.2",
+            "scipy~=1.11.4",
+            "scikit-learn>=1.3.0, <=1.3.2",
         ],
     )