fix test_replace_string_by_dynamic

xadupre · xadupre · commit cd8c2990a066 · 2025-09-29T18:24:01.000+02:00
diff --git a/_unittests/ut_helpers/test_torch_helper.py b/_unittests/ut_helpers/test_torch_helper.py
@@ -240,6 +240,7 @@ def test_replace_string_by_dynamic(self):
                 "_DimHint(type=<_DimHintType.DYNAMIC:3>,min=None,max=None,_factory=True)",
                 "DYN",
             )
+            .replace("DimHint(DYNAMIC)", "DYN")
         )
         self.assertEqual(
             "{'input_ids':{0:DYN,1:DYN},'attention_mask':({0:DYN,1:DYN},),'position_ids':[{0:DYN,1:DYN}]}",

Original file line number	Diff line number	Diff line change
`@@ -240,6 +240,7 @@ def test_replace_string_by_dynamic(self):`
`240`	`240`	`"_DimHint(type=<_DimHintType.DYNAMIC:3>,min=None,max=None,_factory=True)",`
`241`	`241`	`"DYN",`
`242`	`242`	`)`
	`243`	`+ .replace("DimHint(DYNAMIC)", "DYN")`
`243`	`244`	`)`
`244`	`245`	`self.assertEqual(`
`245`	`246`	`"{'input_ids':{0:DYN,1:DYN},'attention_mask':({0:DYN,1:DYN},),'position_ids':[{0:DYN,1:DYN}]}",`