chore(//py/torch_tensorrt): more _Input.py mypy stuff

narendasan · narendasan · commit 80818a9cb815 · 2023-07-19T11:16:57.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/py/torch_tensorrt/_Input.py b/py/torch_tensorrt/_Input.py
@@ -403,5 +403,5 @@ def example_tensor(self, optimization_profile_field: Optional[str] = None) -> Op
                 raise ValueError(
                     "Requested an example tensor from a dynamic shaped input but did not specific which profile field to use."
                 )
-
+        return None
 

Original file line number	Diff line number	Diff line change
`@@ -403,5 +403,5 @@ def example_tensor(self, optimization_profile_field: Optional[str] = None) -> Op`
`403`	`403`	`raise ValueError(`
`404`	`404`	`"Requested an example tensor from a dynamic shaped input but did not specific which profile field to use."`
`405`	`405`	`)`
`406`		`-`
	`406`	`+ return None`
`407`	`407`