ci: run integrations

Author: Teagan42

.github/workflows/ci.yml

@@ -18,5 +18,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements-dev.txt
-      - name: Run tests
+      - name: Run tests (skip GPU sampling)
+        env:
+          LLM_BENCH_SKIP_GPU: '1'
         run: make test

benches/monitor.py

@@ -11,16 +11,19 @@
     psutil = None  # type: ignore
     _PS_OK = False
 
-try:
-    import pynvml
+_NV_OK = False
+pynvml = None
+if not (os.environ.get('LLM_BENCH_SKIP_GPU') == '1'):
     try:
-        pynvml.nvmlInit()
-        _NV_OK = True
+        import pynvml
+        try:
+            pynvml.nvmlInit()
+            _NV_OK = True
+        except Exception:
+            _NV_OK = False
     except Exception:
+        pynvml = None  # type: ignore
         _NV_OK = False
-except Exception:
-    pynvml = None  # type: ignore
-    _NV_OK = False
 
 
 @dataclass