test_nvidia_transform: fix variable reference (#1000)

`out_order` is the global parametrization list, not the test fixture argument
bitsandbytes-foundation · Feb 1, 2024 · 1a0dc5c · 1a0dc5c
1 parent b90db7e
commit 1a0dc5c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)