Merge pull request #132 from dianna-ai/fix-notebook-tests

egpbos · web-flow · commit bb2858c857c4 · 2022-01-25T10:01:36.000+01:00
add notebooks extras_require for running notebooks
diff --git a/.github/actions/install-python-and-package/action.yml b/.github/actions/install-python-and-package/action.yml
@@ -23,4 +23,4 @@ runs:
       shell: bash {0}
       run: |
         python3 -m pip install --upgrade pip setuptools
-        python3 -m pip install .[dev,publishing]
+        python3 -m pip install .[dev,publishing,notebooks]
diff --git a/setup.cfg b/setup.cfg
@@ -69,6 +69,8 @@ dev =
 publishing =
     twine
     wheel
+notebooks =
+    torchvision
 
 [options.packages.find]
 include = dianna, dianna.*
