[fix] add torchvision and umap-learn deps

neurocode-ai · Nov 19, 2023 · 6fce008 · 6fce008
1 parent 01e9c4c
commit 6fce008
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,7 +43,9 @@ matplotlib = "^3.8.2"
 tqdm = "^4.66.1"
 scikit-learn = "^1.3.2"
 pillow = "^10.1.0"
+umap-learn = "^0.5.0"
 torch = { version = "2.0.1+cpu", source = "torch-cpu" }
+torchvision = { version = "0.16.1+cpu", source = "torch-cpu" }
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.11.0"