Update script.py

script.py

@@ -123,18 +123,18 @@ if __name__ == "__main__":
                         })
         
     print('------------ Saving results ---------------')
-    # sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
-    # sub.to_parquet("submission.parquet", index=False)
+    sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
+    sub.to_parquet("submission.parquet", index=False)
 
-    import pyarrow as pa
-    import pyarrow.parquet as pq
+    # import pyarrow as pa
+    # import pyarrow.parquet as pq
     
-    table = pa.table({
-        "order_id": [s["order_id"] for s in solution],
-        "wf_vertices": [s["wf_vertices"] for s in solution],
-        "wf_edges": [s["wf_edges"] for s in solution],
-    })
+    # table = pa.table({
+    #     "order_id": [s["order_id"] for s in solution],
+    #     "wf_vertices": [s["wf_vertices"] for s in solution],
+    #     "wf_edges": [s["wf_edges"] for s in solution],
+    # })
     
-    pq.write_table(table, "submission.parquet")
+    # pq.write_table(table, "submission.parquet")
 
     print("------------ Done ------------ ")