fix: fix assert

ChenZiHong-Gavin · ChenZiHong-Gavin · commit eecc271d3dee · 2025-09-30T17:26:06.000+08:00
diff --git a/graphgen/generate.py b/graphgen/generate.py
@@ -83,7 +83,7 @@ def main():
             )
             assert (
                 config["partition"]["method"] == "ece"
-                and "ece_params" in config["partition"]
+                and "method_params" in config["partition"]
             ), "Only ECE partition with edge sampling is supported."
             config["partition"]["method_params"]["edge_sampling"] = "random"
     elif mode == "cot":

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,7 @@ def main():`
`83`	`83`	`)`
`84`	`84`	`assert (`
`85`	`85`	`config["partition"]["method"] == "ece"`
`86`		`- and "ece_params" in config["partition"]`
	`86`	`+ and "method_params" in config["partition"]`
`87`	`87`	`), "Only ECE partition with edge sampling is supported."`
`88`	`88`	`config["partition"]["method_params"]["edge_sampling"] = "random"`
`89`	`89`	`elif mode == "cot":`