Update script.py

script.py

@@ -121,6 +121,9 @@ if __name__ == "__main__":
             solution.append(res)
 
     print('------------ Saving results ---------------')
-    sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
-    sub.to_parquet("submission.parquet")
+    # sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
+    # sub.to_parquet("submission.parquet")
+    with open("submission.json", "w") as f:
+        json.dump(solution, f)
+
     print("------------ Done ------------ ")