Spaces:

chenzihong
/

GraphGen

Running

github-actions[bot] commited on Sep 30

Commit

1f2cf0b

1 Parent(s): b69388d

Auto-sync from demo at Tue Sep 30 09:26:48 UTC 2025

Files changed (1) hide show

graphgen/generate.py CHANGED Viewed

@@ -83,7 +83,7 @@ def main():
             )
             assert (
                 config["partition"]["method"] == "ece"
-                and "ece_params" in config["partition"]
             ), "Only ECE partition with edge sampling is supported."
             config["partition"]["method_params"]["edge_sampling"] = "random"
     elif mode == "cot":

             )
             assert (
                 config["partition"]["method"] == "ece"
+                and "method_params" in config["partition"]
             ), "Only ECE partition with edge sampling is supported."
             config["partition"]["method_params"]["edge_sampling"] = "random"
     elif mode == "cot":