[export] Minor fix to locals (pytorch#146955)

Pull Request resolved: pytorch#146955 Approved by: https://github.com/bobrenjc93
titaiwangms · Feb 13, 2025 · 5469e5c · 5469e5c
1 parent 7b4efb4
commit 5469e5c
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/torch/fx/experimental/symbolic_shapes.py b/torch/fx/experimental/symbolic_shapes.py
@@ -6492,11 +6492,8 @@ def go(x: Any) -> Optional[str]:
                 for s in x.node.expr.free_symbols:
                     if str(s) in frame_symbols:  # type: ignore[operator]
                         continue
-                    frame_symbols[str(s)] = (  # type: ignore[index]
-                        self.var_to_sources[s][0].name()  # type: ignore[assignment]
-                        if s in self.var_to_sources
-                        else None  # unbacked
-                    )
+                    if s in self.var_to_sources:
+                        frame_symbols[str(s)] = self.var_to_sources[s][0].name()  # type: ignore[assignment]
                 return str(x)
             return None