From 1a0dc5c3bbdd78ae7c6a67283596ccadb7951c70 Mon Sep 17 00:00:00 2001 From: Aarni Koskela Date: Thu, 1 Feb 2024 04:06:47 +0200 Subject: [PATCH] test_nvidia_transform: fix variable reference (#1000) `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)