disable patch

xadupre · xadupre · commit b6bbc95519a1 · 2025-10-13T23:45:26.000+02:00
diff --git a/_unittests/ut_tasks/test_tasks_mask_generation.py b/_unittests/ut_tasks/test_tasks_mask_generation.py
@@ -23,7 +23,7 @@ def test_mask_generation(self):
         model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]
         model(**torch_deepcopy(inputs))
         model(**data["inputs2"])
-        with torch_export_patches(patch_transformers=True, verbose=1):
+        with torch_export_patches(patch_torch=False, patch_transformers=True, verbose=1):
             torch.export.export(
                 model, (), kwargs=inputs, dynamic_shapes=use_dyn_not_str(ds), strict=False
             )

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ def test_mask_generation(self):`
`23`	`23`	`model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]`
`24`	`24`	`model(**torch_deepcopy(inputs))`
`25`	`25`	`model(**data["inputs2"])`
`26`		`- with torch_export_patches(patch_transformers=True, verbose=1):`
	`26`	`+ with torch_export_patches(patch_torch=False, patch_transformers=True, verbose=1):`
`27`	`27`	`torch.export.export(`
`28`	`28`	`model, (), kwargs=inputs, dynamic_shapes=use_dyn_not_str(ds), strict=False`
`29`	`29`	`)`