diff --git a/requirements.txt b/requirements.txt index 7dd6894..308215e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ requests~=2.26.0 requests-html gapandas -sklearn~=0.0 +#sklearn~=0.0 lifetimes~=0.11.3 transformers~=4.5.1 torch @@ -12,4 +12,4 @@ bs4~=0.0.1 beautifulsoup4~=4.9.3 numpy~=1.20.1 scikit-learn~=0.24.1 -setuptools~=45.2.0 \ No newline at end of file +setuptools~=45.2.0 diff --git a/setup.py b/setup.py index 12f2b89..47cecf8 100644 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ ], install_requires=['pandas', 'gapandas', - 'sklearn', + 'scikit-learn', 'requests', 'requests_html', 'httplib2 >= 0.15.0',