update

xadupre · xadupre · commit 5cb500620684 · 2025-03-28T15:29:45.000+01:00
diff --git a/_unittests/ut_xrun_doc/test_torch_test_helper.py b/_unittests/ut_xrun_doc/test_torch_test_helper.py
@@ -93,12 +93,12 @@ def test_replace_string_by_dynamic(self):
         proc = replace_string_by_dynamic(example)
         sproc = (
             str(proc)
-            .replace("_DimHint(type=<_DimHintType.DYNAMIC: 3>", "DYN")
+            .replace("_DimHint(type=<_DimHintType.DYNAMIC: 3>)", "DYN")
             .replace(" ", "")
             .replace("<_DimHint.DYNAMIC:3>", "DYN")
         )
         self.assertEqual(
-            "{'input_ids':{0:DYN),1:DYN)},'attention_mask':({0:DYN),1:DYN)},),'position_ids':[{0:DYN),1:DYN)}]}",
+            "{'input_ids':{0:DYN,1:DYN},'attention_mask':({0:DYN,1:DYN},),'position_ids':[{0:DYN,1:DYN}]}",
             sproc,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -93,12 +93,12 @@ def test_replace_string_by_dynamic(self):`
`93`	`93`	`proc = replace_string_by_dynamic(example)`
`94`	`94`	`sproc = (`
`95`	`95`	`str(proc)`
`96`		`- .replace("_DimHint(type=<_DimHintType.DYNAMIC: 3>", "DYN")`
	`96`	`+ .replace("_DimHint(type=<_DimHintType.DYNAMIC: 3>)", "DYN")`
`97`	`97`	`.replace(" ", "")`
`98`	`98`	`.replace("<_DimHint.DYNAMIC:3>", "DYN")`
`99`	`99`	`)`
`100`	`100`	`self.assertEqual(`
`101`		`- "{'input_ids':{0:DYN),1:DYN)},'attention_mask':({0:DYN),1:DYN)},),'position_ids':[{0:DYN),1:DYN)}]}",`
	`101`	`+ "{'input_ids':{0:DYN,1:DYN},'attention_mask':({0:DYN,1:DYN},),'position_ids':[{0:DYN,1:DYN}]}",`
`102`	`102`	`sproc,`
`103`	`103`	`)`
`104`	`104`