diff --git a/tests/test_image_transforms.py b/tests/test_image_transforms.py index c38a58a1..ea1080f9 100644 --- a/tests/test_image_transforms.py +++ b/tests/test_image_transforms.py @@ -51,7 +51,7 @@ def fixture_single_transforms(): return load_file(ARTIFACT_DIR / "single_transforms.safetensors") -@pytest.fixture(name="single_transforms") +@pytest.fixture(name="img_tensor") def fixture_img_tensor(single_transforms): return single_transforms["original_frame"]