fix: deprecated sklearn name

dwahdany · Dec 13, 2023 · ade6976 · ade6976
1 parent 0a44bc3
commit ade6976
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ version = "0.0.1"
 dependencies = [
   "numpy",
   "torch"
-  "sklearn",
+  "scikit-learn",
 ]
 requires-python = ">=3.8"
 authors = [

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 numpy
 torch
-sklearn
+scikit-learn