[REQ] Install torchvision when building the docs (#49)

f-dangel · web-flow · commit 19ab985d97db · 2023-10-31T14:43:41.000-04:00
diff --git a/setup.cfg b/setup.cfg
@@ -77,6 +77,7 @@ doc =
     mkdocstrings[python]==0.22.0
     mkdocs-gallery==0.7.8
     matplotlib # structure visualizations
+    torchvision # MNIST
 
 # Dependencies needed to run fine-tuning experiments
 fine_tuning =
