diff --git a/_doc/conf.py b/_doc/conf.py index 9f895ec5..77daebca 100644 --- a/_doc/conf.py +++ b/_doc/conf.py @@ -107,6 +107,7 @@ def linkcode_resolve(domain, info): nitpicky = True # See also scikit-learn/scikit-learn#26761 nitpick_ignore = [ + ("py:class", "ast.Node"), ("py:class", "dtype"), ("py:class", "False"), ("py:class", "True"), diff --git a/onnx_diagnostic/torch_export_patches/patch_module.py b/onnx_diagnostic/torch_export_patches/patch_module.py index 0edadab3..7912ebf6 100644 --- a/onnx_diagnostic/torch_export_patches/patch_module.py +++ b/onnx_diagnostic/torch_export_patches/patch_module.py @@ -400,6 +400,8 @@ def transform_method( .. runpython:: :showcode: + :process: + :store_in_file: test_example_transform_method_1.py import torch from onnx_diagnostic.torch_export_patches.patch_module import transform_method @@ -430,6 +432,8 @@ def forward(self, x, y): .. runpython:: :showcode: + :process: + :store_in_file: test_example_transform_method_2.py import torch from onnx_diagnostic.torch_export_patches.patch_module import transform_method