aladac/tensors
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

style: auto-fix ruff warnings [skip ci]

Browse Source
This commit is contained in:
github-actions[bot]
2026-05-17 16:39:54 +00:00
parent 6935491081
commit 592b6858eb
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_tensors.py
+13 -13
View File
@@ -1165,9 +1165,9 @@ class TestValidateModelAvailable:
def test_unknown_model_in_checkpoints_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_unknown_model_in_checkpoints_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""Unknown model + fuzzy-match candidates — exits 1 with did-you-mean.""" """Unknown model + fuzzy-match candidates — exits 1 with did-you-mean."""
import typer # noqa: PLC0415 import typer
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
monkeypatch.setattr( monkeypatch.setattr(
"tensors.comfyui.get_loaded_models", "tensors.comfyui.get_loaded_models",
@@ -1185,9 +1185,9 @@ class TestValidateModelAvailable:
def test_unknown_model_in_diffusion_models_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_unknown_model_in_diffusion_models_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""flux_unet family looks in diffusion_models/, not checkpoints/.""" """flux_unet family looks in diffusion_models/, not checkpoints/."""
import typer # noqa: PLC0415 import typer
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
monkeypatch.setattr( monkeypatch.setattr(
"tensors.comfyui.get_loaded_models", "tensors.comfyui.get_loaded_models",
@@ -1204,7 +1204,7 @@ class TestValidateModelAvailable:
def test_flux2_klein_uses_diffusion_models_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_flux2_klein_uses_diffusion_models_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""flux2_klein family also routes to diffusion_models/.""" """flux2_klein family also routes to diffusion_models/."""
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
monkeypatch.setattr( monkeypatch.setattr(
"tensors.comfyui.get_loaded_models", "tensors.comfyui.get_loaded_models",
@@ -1221,7 +1221,7 @@ class TestValidateModelAvailable:
def test_present_model_passes_silently(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_present_model_passes_silently(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""Happy path — model present, no exception.""" """Happy path — model present, no exception."""
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
monkeypatch.setattr( monkeypatch.setattr(
"tensors.comfyui.get_loaded_models", "tensors.comfyui.get_loaded_models",
@@ -1235,9 +1235,9 @@ class TestValidateModelAvailable:
def test_missing_lora_raises(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_missing_lora_raises(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""Model present but LoRA missing — exit 1.""" """Model present but LoRA missing — exit 1."""
import typer # noqa: PLC0415 import typer
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
monkeypatch.setattr( monkeypatch.setattr(
"tensors.comfyui.get_loaded_models", "tensors.comfyui.get_loaded_models",
@@ -1254,7 +1254,7 @@ class TestValidateModelAvailable:
def test_network_failure_is_non_fatal(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_network_failure_is_non_fatal(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""If get_loaded_models() raises, validation falls through silently.""" """If get_loaded_models() raises, validation falls through silently."""
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
def _boom(console=None): def _boom(console=None):
raise ConnectionError("comfyui down") raise ConnectionError("comfyui down")
@@ -1265,9 +1265,9 @@ class TestValidateModelAvailable:
def test_symlink_hint_when_file_in_wrong_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None: def test_symlink_hint_when_file_in_wrong_bucket(self, monkeypatch: pytest.MonkeyPatch) -> None:
"""flux_unet checkpoint sitting in checkpoints/ → suggest symlinking.""" """flux_unet checkpoint sitting in checkpoints/ → suggest symlinking."""
import typer # noqa: PLC0415 import typer
from tensors import cli as cli_module # noqa: PLC0415 from tensors import cli as cli_module
monkeypatch.setattr( monkeypatch.setattr(
"tensors.comfyui.get_loaded_models", "tensors.comfyui.get_loaded_models",
@@ -1290,9 +1290,9 @@ class TestValidateModelAvailable:
validator's flux_unet / flux2_klein bucket lookup would silently return validator's flux_unet / flux2_klein bucket lookup would silently return
an empty list. an empty list.
""" """
import inspect # noqa: PLC0415 import inspect
import tensors.comfyui as comfyui_module # noqa: PLC0415 import tensors.comfyui as comfyui_module
src = inspect.getsource(comfyui_module.get_loaded_models) src = inspect.getsource(comfyui_module.get_loaded_models)
assert '"diffusion_models"' in src assert '"diffusion_models"' in src