diff --git a/VAME.yaml b/VAME.yaml index 288291a13..609c85c96 100644 --- a/VAME.yaml +++ b/VAME.yaml @@ -20,7 +20,7 @@ dependencies: - pathlib - pandas - ruamel.yaml - - sklearn + - scikit-learn - pyyaml - opencv-python-headless - h5py diff --git a/setup.py b/setup.py index 6cb8cdc29..5b805bec3 100644 --- a/setup.py +++ b/setup.py @@ -24,7 +24,7 @@ "pathlib", "pandas", "ruamel.yaml", - "sklearn", + "scikit-learn", "pyyaml", "opencv-python", ],