adding requirements

dlmbl · Aug 17, 2024 · a7245dc · a7245dc
1 parent 3bbc909
commit a7245dc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/.github/workflows/build-notebooks.yaml b/.github/workflows/build-notebooks.yaml
@@ -18,7 +18,7 @@ jobs:
         run: |
           python -m pip install -U pip
           python -m pip install jupytext nbconvert
-
+          python -m pip install -r requirements.txt
 
       - name: Build notebooks
         run: |

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,11 @@
+ipykernel
+ipywidgets
+torchvision
+tqdm
+scipy
+scikit-learn
+pandas
+seaborn
+git+https://github.com/adjavon/classification.git
+git+https://github.com/dlmbl/dlmbl-unet.git
+git+https://github.com/pytorch/captum.git