docs: Adding words to the refit and engine caching tutorials

Signed-off-by: Naren Dasan <[email protected]>
pytorch · Sep 3, 2024 · 8492c62 · 8492c62
1 parent f9acc9a
commit 8492c62
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docsrc/conf.py b/docsrc/conf.py
@@ -93,7 +93,7 @@
 sphinx_gallery_conf = {
     "examples_dirs": "../examples",
     "gallery_dirs": "tutorials/_rendered_examples/",
-    "ignore_pattern": "utils.py"
+    "ignore_pattern": "utils.py",
 }
 
 # Setup the breathe extension

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: