oops fix test

neuralmagic · Mar 15, 2024 · 013d17b · 013d17b
1 parent 1096700
commit 013d17b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/sparseml/transformers/compression/test_bitmask.py b/tests/sparseml/transformers/compression/test_bitmask.py
@@ -58,7 +58,7 @@ def test_bitmask_sizes(shape, sparsity, dtype):
 
 
 @pytest.mark.parametrize(
-    "shape,sparsity",
+    "shape,sparsity,dtype",
     [
         [(256, 512), 0.5, torch.float32],
         [(128, 280), 0.8, torch.float32],