:build: add pytorch, torchvision to dependencies

gchq · Oct 23, 2024 · 1df2237 · 1df2237
1 parent e94aed0
commit 1df2237
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.cspell/library_terms.txt b/.cspell/library_terms.txt
@@ -101,6 +101,7 @@ pyqt
 pyright
 pyroma
 pytest
+pytorch
 pytree
 pytrees
 quickstart
@@ -109,7 +110,6 @@ rcond
 rect
 rerunfailures
 rightarrow
-rngs
 rtol
 samp
 scikit
@@ -124,6 +124,7 @@ tensordot
 texttt
 toctree
 tomli
+torchvision
 tqdm
 triu
 ttest

diff --git a/pyproject.toml b/pyproject.toml
@@ -34,6 +34,8 @@ dependencies = [
     "scikit-learn",
     "tqdm",
     "typing-extensions",
+    "pytorch",
+    "torchvision"
 ]
 
 [project.optional-dependencies]