Update the regex for capturing the partition stats

sirakiin · copybara-github · commit 13a29dc756c6 · 2025-11-20T18:50:54.000-08:00
LiteRT-PiperOrigin-RevId: 835000723
diff --git a/litert/python/aot/core/apply_plugin.py b/litert/python/aot/core/apply_plugin.py
@@ -29,8 +29,8 @@
 _BINARY = pathlib.Path("tools/apply_plugin_main")
 
 _RE_PARTITION_STATS = re.compile(
-    r"PartitionSubgraph: (\d+), selected num ops: (\d+), from totoal ops:"
-    r" (\d+), num partitions: (\d+)"
+    r"Partitioned subgraph<(\d+)>, selected (\d+) ops, from a total of "
+    r"(\d+) ops. resulted in (\d+) partitions."
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -29,8 +29,8 @@`
`29`	`29`	`_BINARY = pathlib.Path("tools/apply_plugin_main")`
`30`	`30`
`31`	`31`	`_RE_PARTITION_STATS = re.compile(`
`32`		`- r"PartitionSubgraph: (\d+), selected num ops: (\d+), from totoal ops:"`
`33`		`- r" (\d+), num partitions: (\d+)"`
	`32`	`+ r"Partitioned subgraph<(\d+)>, selected (\d+) ops, from a total of "`
	`33`	`+ r"(\d+) ops. resulted in (\d+) partitions."`
`34`	`34`	`)`
`35`	`35`
`36`	`36`