Commit 9001e5ed authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): fix bug in torch

parent a1e37102
Loading
Loading
Loading
Loading

test/conftest.py

0 → 100644
+11 −0
Original line number Diff line number Diff line
import pytest

try:
    import torch
except (ImportError, ModuleNotFoundError):
    pass


@pytest.fixture(autouse=True)
def _try_import_torch():
    yield
+0 −3
Original line number Diff line number Diff line
import pytest
import torch

from imgutils.tagging import get_deepdanbooru_tags
from test.testings import get_testfile

_ = torch


@pytest.mark.unittest
class TestTaggingDeepdanbooru:
+0 −3
Original line number Diff line number Diff line
import pytest
import torch

from imgutils.tagging import get_wd14_tags
from test.testings import get_testfile

_ = torch


@pytest.mark.unittest
class TestTaggingWd14:
+0 −3
Original line number Diff line number Diff line
import os.path

import pytest
import torch
from hbutils.testing import tmatrix

from imgutils.validate.monochrome import get_monochrome_score, is_monochrome

_ = torch


def get_samples():
    return [