Merge pull request #309 from not522/skorch-ci

y0z · web-flow · commit 70be1317109a · 2025-03-24T14:25:30.000+09:00
Run the skorch example in the CI
diff --git a/.github/workflows/pytorch.yml b/.github/workflows/pytorch.yml
@@ -77,3 +77,8 @@ jobs:
         python pytorch/pytorch_lightning_ddp.py -p
       env:
         OMP_NUM_THREADS: 1
+    - name: Run skorch example
+      run: |
+        python pytorch/skorch_simple.py
+      env:
+        OMP_NUM_THREADS: 1
diff --git a/pytorch/requirements.txt b/pytorch/requirements.txt
@@ -1,4 +1,5 @@
 mpi4py
+pandas
 plotly
 pytorch-ignite
 lightning
