diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index d9f56dc..26d6f1a 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -24,6 +24,9 @@ jobs: - name: poetry install run: poetry install --no-interaction --extras=training + - name: black + run: poetry run black --check . + - name: lint run: poetry run ruff check . diff --git a/tests/conftest.py b/tests/conftest.py index cf7a28b..decf97f 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -24,6 +24,7 @@ def test_weights_path() -> Path: def test_e2e_path() -> Path: return PARENT_PATH / "e2e" + @fixture(scope="session") def test_textual_inversion_path() -> Path: return PARENT_PATH / "foundationals" / "clip" / "test_concepts_ref" diff --git a/tests/foundationals/clip/test_concepts.py b/tests/foundationals/clip/test_concepts.py index 6df3e07..aed634d 100644 --- a/tests/foundationals/clip/test_concepts.py +++ b/tests/foundationals/clip/test_concepts.py @@ -41,8 +41,10 @@ def our_encoder_with_new_concepts( @pytest.fixture(scope="module") -def ref_sd15_with_new_concepts(runwayml_weights_path: Path, test_textual_inversion_path: Path, test_device: torch.device): - pipe = StableDiffusionPipeline.from_pretrained(runwayml_weights_path, torch_dtype=torch.float16).to(test_device) # type: ignore +def ref_sd15_with_new_concepts( + runwayml_weights_path: Path, test_textual_inversion_path: Path, test_device: torch.device +): + pipe = StableDiffusionPipeline.from_pretrained(runwayml_weights_path, torch_dtype=torch.float16).to(test_device) # type: ignore pipe.load_textual_inversion(test_textual_inversion_path / "cat-toy") # type: ignore pipe.load_textual_inversion(test_textual_inversion_path / "gta5-artwork") # type: ignore return pipe