imaginAIry/tests/conftest.py

import sys

import pytest

from imaginairy import api
from imaginairy.suppress_logs import suppress_annoying_logs_and_warnings

if "pytest" in str(sys.argv):
    suppress_annoying_logs_and_warnings()


@pytest.fixture(scope="session", autouse=True)
def pre_setup():
    api.IMAGINAIRY_SAFETY_MODE = "disabled"
refactor/test: logging suppression + hashed image test - simpler logging suppression for `transformers` library - suppress logging noise for running tests - get test running for all samplers on mps and cuda platforms - refactor safety model env variable to allow classification 2022-09-15 02:40:50 +00:00			`import sys`

			`import pytest`

feature: urls as init images - --init-image accepts urls - cleanup command line code 2022-09-16 06:06:59 +00:00			`from imaginairy import api`
refactor/test: logging suppression + hashed image test - simpler logging suppression for `transformers` library - suppress logging noise for running tests - get test running for all samplers on mps and cuda platforms - refactor safety model env variable to allow classification 2022-09-15 02:40:50 +00:00			`from imaginairy.suppress_logs import suppress_annoying_logs_and_warnings`

			`if "pytest" in str(sys.argv):`
			`suppress_annoying_logs_and_warnings()`


			`@pytest.fixture(scope="session", autouse=True)`
			`def pre_setup():`
			`api.IMAGINAIRY_SAFETY_MODE = "disabled"`