fix test_replace_string_by_dynamic for torch nightly build (#243)

xadupre · web-flow · commit 7e35e7f765c9 · 2025-09-29T18:40:57.000+02:00
* Simple change to check CI

* fix test_replace_string_by_dynamic
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}]}",
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+ml_dtypes
 numpy
 onnx>=1.16.0
 onnxruntime>=1.23

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}]}",`