fix tests

dsikka · dsikka · commit 30ad30599326 · 2025-07-30T14:13:15.000Z
diff --git a/tests/test_quantization/test_utils/test_helpers.py b/tests/test_quantization/test_utils/test_helpers.py
@@ -83,7 +83,7 @@ def test_fused_global_scales():
     "shape,group_size,exp_shape",
     [
         # Only batch size =1 is supported for dynamic GROUP quantization
-        ((1, 4, 8), 4, torch.Size([4, 2])),
+        ((1, 4, 8), 4, torch.Size([1, 4, 2])),
     ],
 )
 def test_compute_dynamic_scales_and_zp_group(shape, group_size, exp_shape):

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,7 @@ def test_fused_global_scales():`
`83`	`83`	`"shape,group_size,exp_shape",`
`84`	`84`	`[`
`85`	`85`	`# Only batch size =1 is supported for dynamic GROUP quantization`
`86`		`- ((1, 4, 8), 4, torch.Size([4, 2])),`
	`86`	`+ ((1, 4, 8), 4, torch.Size([1, 4, 2])),`
`87`	`87`	`],`
`88`	`88`	`)`
`89`	`89`	`def test_compute_dynamic_scales_and_zp_group(shape, group_size, exp_shape):`