Tests: set matplotlib backend locally too (#2326)

microsoft · Sep 28, 2024 · 742b900 · 742b900
1 parent 69f91a2
commit 742b900
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -15,8 +15,6 @@ jobs:
   latest:
     name: latest
     runs-on: ${{ matrix.os }}
-    env:
-      MPLBACKEND: Agg
     strategy:
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
@@ -55,8 +53,6 @@ jobs:
   minimum:
     name: minimum
     runs-on: ubuntu-latest
-    env:
-      MPLBACKEND: Agg
     steps:
       - name: Clone repo
         uses: actions/[email protected]
@@ -90,8 +86,6 @@ jobs:
   datasets:
     name: datasets
     runs-on: ubuntu-latest
-    env:
-      MPLBACKEND: Agg
     steps:
       - name: Clone repo
         uses: actions/[email protected]

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -4,6 +4,7 @@
 from pathlib import Path
 from typing import Any
 
+import matplotlib
 import pytest
 import torch
 import torchvision
@@ -19,6 +20,11 @@ def load_state_dict_from_url(monkeypatch: MonkeyPatch) -> None:
     monkeypatch.setattr(torchvision.models._api, 'load_state_dict_from_url', load)
 
 
+@pytest.fixture(autouse=True, scope='session')
+def matplotlib_backend() -> None:
+    matplotlib.use('agg')
+
+
 @pytest.fixture(autouse=True)
 def torch_hub(tmp_path: Path) -> None:
     torch.hub.set_dir(tmp_path)  # type: ignore[no-untyped-call]