diff --git a/requirements.txt b/requirements.txt index 0e481a9..724f840 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ matplotlib>=3.1.1 monty>=2.0.4 numpy==1.19.4 pandas==1.1.4 -scikit-learn==0.23.2 +scikit-learn==1.0.1 scipy>=1.3.0 seaborn>=0.9.0 matminer==0.6.3 diff --git a/setup.py b/setup.py index ed30e34..c8a3b0f 100644 --- a/setup.py +++ b/setup.py @@ -15,7 +15,7 @@ url="https://github.com/ncfrey/pumml", packages=setuptools.find_packages(), python_requires='>=3.6', - install_requires=['matplotlib>=3.1.1', 'monty>=2.0.4', 'numpy==1.19.4', 'pandas==1.1.4', 'scikit-learn==0.23.2', 'scipy>=1.3.0', 'seaborn>=0.9.0', + install_requires=['matplotlib>=3.1.1', 'monty>=2.0.4', 'numpy==1.19.4', 'pandas==1.1.4', 'scikit-learn==1.0.1', 'scipy>=1.3.0', 'seaborn>=0.9.0', 'matminer==0.6.3', 'pymatgen==2020.11.11'], keywords=["VASP", "machine", "learning", "materials", "science", "DFT", "synthesis"],