mo

Author: xadupre

CHANGELOGS.rst

@@ -4,7 +4,7 @@ Change Logs
 0.6.1
 +++++
 
-* :pr:`115`, :pr:`116`: first steps for TorchEvaluator
+* :pr:`115`, :pr:`116`, :pr:`117`: first steps for TorchEvaluator
 * :pr:`114`: extends the list of known rewritings
 * :pr:`113`: fixes a couple of issues with ModelBuilder
 

onnx_diagnostic/torch_onnx/runtime_info.py

@@ -43,7 +43,7 @@ def __init__(
         name: str,
         dtype: Optional[Any] = None,
         shape: Optional[Tuple[Union[str, int], ...]] = None,
-        value: Optional[torch.Tensor] = None,
+        value: Optional[Any] = None,
         first_used: Optional[int] = None,
         last_used: Optional[int] = None,
         created: Optional[int] = None,