test(conftest): 添加matplotlib后端设置以避免GUI问题
This commit is contained in:
parent
9a452dd34e
commit
85aeec9e55
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Reference in New Issue