fix documentation

Author: xadupre

onnx_diagnostic/export/dynamic_shapes.py

@@ -753,14 +753,18 @@ def change_dynamic_dimensions(self):
             :showcode:
 
             import torch
+            from onnx_diagnostic.helpers import string_type
             from onnx_diagnostic.export.dynamic_shapes import CoupleInputsDynamicShapes
 
-            T3x1 = torch.rand((3, 1))
+            T3x15 = torch.rand((3, 15))
+            T3x20 = torch.rand((3, 20))
             T3x4 = torch.rand((3, 4))
             ds_batch = {0: "batch"}
             ds_batch_seq = {0: "batch", 1: "seq"}
-            kwargs = {"A": T3x4, "B": (T3x1, T3x1)}
+            kwargs = {"A": T3x4, "B": (T3x15, T3x20)}
             ds = {"A": ds_batch, "B": (ds_batch, ds_batch_seq)}
-            print(CoupleInputsDynamicShapes((), kwargs, ds).change_dynamic_dimension())
+            new_kwargs = CoupleInputsDynamicShapes((), kwargs, ds).change_dynamic_dimensions()
+            print("before:", string_type(kwargs, with_shape=True))
+            print("-after:", string_type(new_kwargs, with_shape=True))
         """
         return self._generic_walker(self.ChangeDimensionProcessor())

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

@@ -47,7 +47,7 @@ def _patch_make_causal_mask(
 if sys.version_info[:2] <= (3, 11):
 
     @dataclass
-    class kkpatched_AttentionMaskConverter:
+    class patched_AttentionMaskConverter:
         """
         Patches
         ``transformers.modeling_attn_mask_utils.AttentionMaskConverter._make_causal_mask``.
@@ -72,7 +72,7 @@ def _make_causal_mask(
 else:
 
     @dataclass
-    class kkpatched_AttentionMaskConverter:
+    class patched_AttentionMaskConverter:
         """
         Patches
         ``transformers.modeling_attn_mask_utils.AttentionMaskConverter._make_causal_mask``.

onnx_diagnostic/torch_models/test_helper.py

@@ -277,8 +277,7 @@ def validate_model(
         if verbose:
             print(f"[validate_model] new inputs: {string_type(data['inputs'])}")
             print(
-                f"[validate_model] new dynnamic_shapes: "
-                f"{_ds_clean(data['dynamic_shapes'])}"
+                f"[validate_model] new dynamic_hapes: {_ds_clean(data['dynamic_shapes'])}"
             )
 
     if not empty(dtype):