Attempting build fix by including torchvision in deps

lincc-frameworks · Aug 24, 2024 · 3bffb95 · 3bffb95
1 parent 02d56fb
commit 3bffb95
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,6 +19,7 @@ dependencies = [
     "astropy", # Used to load fits files of sources to query HSC cutout server
     "toml", # Used to load configuration files as dictionaries
     "torch", # Used for CNN model and in train.py
+    "torchvision", # Used in data loader and example autoencoder
 ]
 
 [project.scripts]