groupby

shink · shink · commit 8269a85d5a1e · 2025-04-28T07:56:33.000Z
diff --git a/torch/_dynamo/variables/iter.py b/torch/_dynamo/variables/iter.py
@@ -179,7 +179,8 @@ def retrieve_const_key(key):
                         gb_type="Unsupported key type for itertools.groupby",
                         context=f"call_function {self} {args} {kwargs}",
                         explanation="Dynamo does not know how to trace "
-                        f"itertools.groupby with key type: {str(type(key))}",
+                        f"itertools.groupby with key type: {str(type(key))}. "
+                        "We only support grouping keys that are constants (int, float, str, etc.)",
                         hints=[*graph_break_hints.SUPPORTABLE],
                     )
 

Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,8 @@ def retrieve_const_key(key):`
`179`	`179`	`gb_type="Unsupported key type for itertools.groupby",`
`180`	`180`	`context=f"call_function {self} {args} {kwargs}",`
`181`	`181`	`explanation="Dynamo does not know how to trace "`
`182`		`- f"itertools.groupby with key type: {str(type(key))}",`
	`182`	`+ f"itertools.groupby with key type: {str(type(key))}. "`
	`183`	`+ "We only support grouping keys that are constants (int, float, str, etc.)",`
`183`	`184`	`hints=[*graph_break_hints.SUPPORTABLE],`
`184`	`185`	`)`
`185`	`186`