Hot fixes

.github/workflows/notebook-pr.yaml

@@ -15,7 +15,22 @@ jobs:
 
     runs-on: ubuntu-latest
     steps:
-
+      - name: Free up disk space
+        uses: jlumbroso/free-disk-space@main
+        with:
+          # this might remove tools that are actually needed,
+          # if set to "true" but frees about 6 GB
+          tool-cache: false
+
+          # all of these default to true, but feel free to set to
+          # "false" if necessary for your workflow
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: true
+          docker-images: true
+          swap-storage: true
+
       - name: Checkout
         uses: actions/checkout@v3
         with:
@@ -48,6 +63,9 @@ jobs:
       - name: Install dependencies
         if: "!contains(env.COMMIT_MESSAGE, 'skip ci') && contains(env.COMMIT_MESSAGE, 'ci:execute')"
         run: |
+          # Install dependencies for github processing
+          sudo apt install nvidia-cuda-toolkit -y
+          sudo apt-get update && sudo apt install xvfb -y
           python -m pip install --upgrade pip wheel
           pip install -r requirements.txt
           pip install jupyter_client==7.3.5  # downgrade jupyter-client to fix hangs

requirements.txt

@@ -1,12 +1,17 @@
 requests
-numpy
+numpy==1.26.4
 scipy
 matplotlib
 scikit-learn
-torch
+torch==1.13.0
+torchvision==0.14.0
 ipywidgets
 tqdm
 torchvision
 pathlib
 xkcd
-decorator==5.0.9
+decorator==4.0.2
+pyvirtualdisplay
+tensorboard
+moviepy==1.0.3
+imageio_ffmpeg