fix mypy

Author: xadupre

_unittests/ut_helpers/test_torch_test_helper.py

@@ -204,7 +204,7 @@ def forward(self, x, y):
             else:
                 print("output", k, v)
         print(string_type(restored, with_shape=True))
-        l1, l2 = 151, 160
+        l1, l2 = 182, 191
         self.assertEqual(
             [
                 (f"-Model-{l2}", 0, "I"),

onnx_diagnostic/helpers/torch_test_helper.py

@@ -192,6 +192,7 @@ def forward(self, x, y):
             f[0].forward = f[1]
         if dump_file:
             # Let's add the cached tensor
+            assert storage is not None, "storage cannot be None but mypy is confused here."
             storage.update(_additional_stolen_objects)
             # We clear the cache.
             _additional_stolen_objects.clear()