docs: Adding words to the refit and engine caching tutorials

Signed-off-by: Naren Dasan <[email protected]>
pytorch · Sep 3, 2024 · d5e3a77 · d5e3a77
1 parent f9acc9a
commit d5e3a77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/torch_tensorrt/dynamo/utils.py b/py/torch_tensorrt/dynamo/utils.py
@@ -347,7 +347,7 @@ def unwrap_tensor_shape(
     if isinstance(tensor, int):
         tensor_shape.append(tensor)
     elif isinstance(tensor, torch.SymInt):
-        min_max_opt: Dict[str, int] = extract_var_range_info(tensor)
+        min_max_opt = extract_var_range_info(tensor)
         tensor_shape.append((min_max_opt["min"], min_max_opt["max"]))
     elif isinstance(tensor, (torch.Tensor, FakeTensor)):
         for dimension in tensor.shape: