From b6b078d2bef70d743aa562aa1984eb43c4e7bcdc Mon Sep 17 00:00:00 2001 From: Aarni Koskela Date: Wed, 31 Jan 2024 08:39:42 +0200 Subject: [PATCH] test_nvidia_transform: fix variable reference `out_order` is the global parametrization list, not the test fixture argument --- tests/test_functional.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_functional.py b/tests/test_functional.py index 5b7f83bc3..340278912 100644 --- a/tests/test_functional.py +++ b/tests/test_functional.py @@ -613,9 +613,9 @@ def test_vector_quant(dim1, dim2, dim3): @pytest.mark.parametrize("dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose",values,ids=names) def test_nvidia_transform(dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose): - if dims == 3 and out_order != "col32": + if dims == 3 and orderOut != "col32": return - if dtype == torch.int32 and out_order != "col32": + if dtype == torch.int32 and orderOut != "col32": return try: func = F.get_transform_func(dtype, orderA, orderOut, transpose)