From ad8e28801d0608d4ef66934ad716af67de5a6388 Mon Sep 17 00:00:00 2001 From: Noah Harasz Date: Tue, 19 Dec 2023 18:04:12 -0800 Subject: [PATCH] pin versions to later versions, replace one_hot_encoder feature name getter --- numerai_tools/scoring.py | 2 +- setup.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/numerai_tools/scoring.py b/numerai_tools/scoring.py index c7b735c..92fa731 100644 --- 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 index db75a8c..1124fb7 100644 --- 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", ], )