test(conftest): 添加matplotlib后端设置以避免GUI问题

2025-10-03 22:11:54 +08:00
parent 9a452dd34e
commit 85aeec9e55
1 changed files with 3 additions and 0 deletions
@@ -6,12 +6,15 @@ import shutil
 import torch

 import pytest
+import matplotlib
 from torch.utils.data import Dataset

 from khaosz.core import *
 from khaosz.trainer import *
 from khaosz.trainer.data_util import *

+matplotlib.use("Agg")
+
@pytest.fixture
 def base_test_env():
    test_dir = tempfile.mkdtemp()