train only 1 epoch in cicd and print to stdout

mllam · Aug 14, 2024 · 57bbb81 · 57bbb81
1 parent 799d55e
commit 57bbb81
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci-pdm-install-and-test-cpu.yml b/.github/workflows/ci-pdm-install-and-test-cpu.yml
@@ -47,7 +47,7 @@ jobs:
 
       - name: Run tests
         run: |
-          pdm run pytest
+          pdm run pytest -s tests/
 
       - name: Save cache data
         uses: actions/cache/save@v4

diff --git a/.github/workflows/ci-pdm-install-and-test-gpu.yml b/.github/workflows/ci-pdm-install-and-test-gpu.yml
@@ -52,7 +52,7 @@ jobs:
 
       - name: Run tests
         run: |
-          pdm run pytest
+          pdm run pytest -s tests/
 
       - name: Save cache data
         uses: actions/cache/save@v4

diff --git a/.github/workflows/ci-pip-install-and-test-cpu.yml b/.github/workflows/ci-pip-install-and-test-cpu.yml
@@ -40,7 +40,7 @@ jobs:
 
       - name: Run tests
         run: |
-          python -m pytest
+          python -m pytest -s tests/
 
       - name: Save cache data
         uses: actions/cache/save@v4

diff --git a/.github/workflows/ci-pip-install-and-test-gpu.yml b/.github/workflows/ci-pip-install-and-test-gpu.yml
@@ -42,7 +42,7 @@ jobs:
 
       - name: Run tests
         run: |
-          python -m pytest
+          python -m pytest -s tests/
 
       - name: Save cache data
         uses: actions/cache/save@v4

diff --git a/tests/test_training.py b/tests/test_training.py
@@ -25,7 +25,7 @@ def test_training(datastore_name):
         device_name = "cpu"
 
     trainer = pl.Trainer(
-        max_epochs=3,
+        max_epochs=1,
         deterministic=True,
         accelerator=device_name,
         # XXX: `devices` has to be set to 2 otherwise