ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
ViperEkura 2025-10-03 22:11:54 +08:00
parent 9a452dd34e
commit 85aeec9e55
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/conftest.py
View File

@ -6,12 +6,15 @@ import shutil
import torch import torch
import pytest import pytest
import matplotlib
from torch.utils.data import Dataset from torch.utils.data import Dataset
from khaosz.core import * from khaosz.core import *
from khaosz.trainer import * from khaosz.trainer import *
from khaosz.trainer.data_util import * from khaosz.trainer.data_util import *
matplotlib.use("Agg")
@pytest.fixture @pytest.fixture
def base_test_env(): def base_test_env():
test_dir = tempfile.mkdtemp() test_dir = tempfile.mkdtemp()