mypy

Author: xadupre

onnx_diagnostic/api.py

@@ -1,2 +1,15 @@
+from typing import Any
+
+
 class TensorLike:
     """Mocks a tensor."""
+
+    @property
+    def dtype(self) -> Any:
+        "Must be overwritten."
+        raise NotImplementedError("dtype must be overwritten.")
+
+    @property
+    def shape(self) -> Any:
+        "Must be overwritten."
+        raise NotImplementedError("shape must be overwritten.")

onnx_diagnostic/reference/torch_ops/_op_run.py

@@ -159,6 +159,10 @@ def copy(self) -> "OpRunSequence":
         "Shallow copy."
         return self.__class__(self.sequence, dtype=self.dtype)
 
+    def string_type(self) -> str:
+        "Returns a string which can be printed."
+        return string_type(self.sequence, with_shape=True)
+
 
 class OpRun:
     """