tests: disable test if running on github actions

pull/57/head
Bryce 2 years ago committed by Bryce Drennan
parent b0123a8f43
commit a9bde7386c

@ -1,9 +1,12 @@
import pytest
from PIL import Image
from imaginairy.safety import create_safety_score
from imaginairy.utils import get_device
from tests import TESTS_FOLDER
@pytest.mark.skipif(get_device() == "cpu", reason="Too slow to run on CPU")
def test_is_nsfw():
img = Image.open(f"{TESTS_FOLDER}/data/safety.jpg")

Loading…
Cancel
Save