fix dynamic shape

titaiwangms · titaiwangms · commit 77939dd57627 · 2025-09-26T23:55:06.000Z
diff --git a/_unittests/ut_export/test_dynamic_shapes.py b/_unittests/ut_export/test_dynamic_shapes.py
@@ -859,9 +859,9 @@ def test_unbatch_inputs(self):
         )
         s = self.string_type(new_dims, with_shape=True)
         self.assertEqual(
-            "dict(input_ids:T7s1x3,attention_mask:T7s1x33,position_ids:T7s1x3,"
+            "dict(input_ids:T7s1x1,attention_mask:T7s1x33,position_ids:T7s1x1,"
             "past_key_values:DynamicCache("
-            "key_cache=#1[T1s1x1x30x96], value_cache=#1[T1s1x1x30x96]))",
+            "key_cache=#1[T1s1x1x32x96], value_cache=#1[T1s1x1x32x96]))",
             s,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -859,9 +859,9 @@ def test_unbatch_inputs(self):`
`859`	`859`	`)`
`860`	`860`	`s = self.string_type(new_dims, with_shape=True)`
`861`	`861`	`self.assertEqual(`
`862`		`- "dict(input_ids:T7s1x3,attention_mask:T7s1x33,position_ids:T7s1x3,"`
	`862`	`+ "dict(input_ids:T7s1x1,attention_mask:T7s1x33,position_ids:T7s1x1,"`
`863`	`863`	`"past_key_values:DynamicCache("`
`864`		`- "key_cache=#1[T1s1x1x30x96], value_cache=#1[T1s1x1x30x96]))",`
	`864`	`+ "key_cache=#1[T1s1x1x32x96], value_cache=#1[T1s1x1x32x96]))",`
`865`	`865`	`s,`
`866`	`866`	`)`
`867`	`867`