Update memtests.yml

jcapellman · Sep 20, 2024 · 1407788 · 1407788
1 parent 56e925a
commit 1407788
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/memtests.yml b/.github/workflows/memtests.yml
@@ -19,4 +19,8 @@ jobs:
           cache: pip
       - run: pip install memray
       - run: pip install -r mlids-trainer/requirements.txt
-      - run: python -m memray run mlids-trainer/main.py
+      - run: python -m memray run output=memray_report.bin mlids-trainer/main.py
+      - run: python -m memrary stats memray_report.bin --threshold 1MB
+        continue-on-error: true
+      - if: failure()
+        run: echo "Memory usage exceeded 1MB" && exit 1