Update

XuehaiPan · XuehaiPan · commit c8eb125a303c · 2024-11-21T00:58:31.000+08:00
[ghstack-poisoned]
diff --git a/torch/_dynamo/polyfills/operator.py b/torch/_dynamo/polyfills/operator.py
@@ -100,6 +100,9 @@ def getter(obj: Any) -> tuple[Any, ...]:  # type: ignore[misc]
 # Reference: https://docs.python.org/3/library/operator.html#operator.methodcaller
 @substitute_in_graph(operator.methodcaller, is_embedded_type=True)  # type: ignore[arg-type]
 def methodcaller(name: str, /, *args: Any, **kwargs: Any) -> Callable[[Any], Any]:
+    if not isinstance(name, str):
+        raise TypeError("method name must be a string")
+
     def caller(obj: Any) -> Any:
         return getattr(obj, name)(*args, **kwargs)
 
