From e77c7e40b7981d04f21cbb1ec6258d9c632ce0dc Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 17 May 2024 22:53:45 -0400 Subject: [PATCH] fix ruff error --- tests/app/services/model_load/test_load_api.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/app/services/model_load/test_load_api.py b/tests/app/services/model_load/test_load_api.py index 57d0fed341..2af321d60f 100644 --- a/tests/app/services/model_load/test_load_api.py +++ b/tests/app/services/model_load/test_load_api.py @@ -79,4 +79,6 @@ def test_download_diffusers(mock_context: InvocationContext) -> None: def test_download_diffusers_subfolder(mock_context: InvocationContext) -> None: model_path = mock_context.models.download_and_cache_model("stabilityai/sdxl-turbo::vae") assert model_path.is_dir() - assert (model_path / "diffusion_pytorch_model.fp16.safetensors").exists() or (model_path / "diffusion_pytorch_model.safetensors").exists() + assert (model_path / "diffusion_pytorch_model.fp16.safetensors").exists() or ( + model_path / "diffusion_pytorch_model.safetensors" + ).exists()