Skip to content

Commit 9f93796

Browse files
Lincoln Steinpsychedelicious
authored andcommitted
ruff fixes
1 parent f81b8bc commit 9f93796

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tests/app/services/model_load/test_load_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,13 @@ def test_load_from_path(mock_context: InvocationContext, embedding_file: Path) -
5757
assert isinstance(loaded_model_3.model, dict)
5858
assert torch.equal(loaded_model_1.model["emb_params"], loaded_model_3.model["emb_params"])
5959

60+
6061
def test_load_from_dir(mock_context: InvocationContext, vae_directory: Path) -> None:
6162
loaded_model = mock_context.models.load_and_cache_model(vae_directory)
6263
assert isinstance(loaded_model, LoadedModelWithoutConfig)
6364
assert isinstance(loaded_model.model, AutoencoderTiny)
6465

66+
6567
def test_download_and_load(mock_context: InvocationContext) -> None:
6668
loaded_model_1 = mock_context.models.load_and_cache_model(
6769
"https://www.test.foo/download/test_embedding.safetensors"

tests/backend/model_manager/model_manager_fixtures.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ def mm2_model_files(tmp_path_factory) -> Path:
6060
def embedding_file(mm2_model_files: Path) -> Path:
6161
return mm2_model_files / "test_embedding.safetensors"
6262

63+
6364
@pytest.fixture
6465
def vae_directory(mm2_model_files: Path) -> Path:
6566
return mm2_model_files / "taesdxl"

0 commit comments

Comments
 (0)