From 60d2e32fa4f49208a4f0389c6bb74141534e60db Mon Sep 17 00:00:00 2001 From: "Edward Z. Yang" Date: Mon, 13 Jan 2025 19:38:52 -0800 Subject: [PATCH] [BE] Remove lambda from str (#144743) Signed-off-by: Edward Z. Yang Pull Request resolved: https://github.com/pytorch/pytorch/pull/144743 Approved by: https://github.com/avikchaudhuri, https://github.com/Skylion007 ghstack dependencies: #144471 --- torch/fx/experimental/symbolic_shapes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/fx/experimental/symbolic_shapes.py b/torch/fx/experimental/symbolic_shapes.py index 3b1edea9083..eefeb948b69 100644 --- a/torch/fx/experimental/symbolic_shapes.py +++ b/torch/fx/experimental/symbolic_shapes.py @@ -5628,7 +5628,7 @@ class ShapeEnv: s in self.size_like, self.oblivious_upper_bound_exclusive.get(s), ) - for s in sorted(fs, key=lambda s: str(s)) # TODO: speed up sort? + for s in sorted(fs, key=str) # TODO: speed up sort? ) r = _maybe_evaluate_static_worker(