fix: scikit-learn requirement

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

Hide file tree

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

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,3 @@
 numpy
-torch
+torch
+sklearn
-Original file line number
+Diff line change
@@ -1,2 +1,3 @@
     numpy
-    torch
+    torch
+    sklearn