fix

Author: xadupre

_doc/conf.py

@@ -127,6 +127,7 @@
     ("py:func", "torch._export.tools.report_exportability"),
     ("py:meth", "huggingface_hub.HfApi.list_models"),
     ("py:meth", "transformers.GenerationMixin.generate"),
+    ("py:meth", "unittests.TestCase.subTest"),
 ]
 
 nitpick_ignore_regex = [

_unittests/ut_torch_export_patches/test_onnx_export_errors.py

@@ -37,7 +37,6 @@ def __init__(self):
         with bypass_export_some_errors(verbose=1):
             values, spec = py_pytree.tree_flatten(cache)
             cache2 = py_pytree.tree_unflatten(values, spec)
-            self.assertEqual(cache.dtype, cache2.dtype)
             self.assertEqual(cache.max_batch_size, cache2.max_batch_size)
             self.assertEqual(cache.intermediate_size, cache2.intermediate_size)
             self.assertEqual(cache.ssm_state_size, cache2.ssm_state_size)