add none

Author: xadupre

CHANGELOGS.rst

@@ -1,6 +1,11 @@
 Change Logs
 ===========
 
+0.4.2
++++++
+
+* :pr:`73`: supports MambaCache in max_diff, torch_deepcopy
+
 0.4.1
 +++++
 

_unittests/ut_helpers/test_torch_test_helper.py

@@ -151,6 +151,9 @@ def test_torch_deepcopy_sliding_windon_cache(self):
         hash2 = string_type(at, with_shape=True, with_min_max=True)
         self.assertEqual(hash1, hash2)
 
+    def test_torch_deepcopy_none(self):
+        self.assertEmpty(torch_deepcopy(None))
+
 
 if __name__ == "__main__":
     unittest.main(verbosity=2)

onnx_diagnostic/helpers/torch_test_helper.py

@@ -347,6 +347,8 @@ def torch_deepcopy(value: Any) -> Any:
     """
     Makes a deepcopy.
     """
+    if value is None:
+        return None
     if isinstance(value, (int, float, str)):
         return value
     if isinstance(value, tuple):