move to json

params.json

@@ -18,6 +18,6 @@
     "time_limit": 7200,
     "dataset": "usm3d/hoho22k_2026_test_x_anon",
     "submission_filenames": [
-        "submission.parquet"
+        "submission.json"
     ]
 }

script.py

@@ -2,12 +2,12 @@
 
 ### You can change the rest of the code to define and test your solution.
 ### However, you should not change the signature of the provided function.
-### The script saves "submission.parquet" file in the current directory.
+### The script saves "submission.json" file in the current directory.
 ### You can use any additional files and subdirectories to organize your code.
 
 from pathlib import Path
 from tqdm import tqdm
-import pandas as pd
+import json
 import numpy as np
 from datasets import load_dataset
 from typing import Dict
@@ -123,18 +123,7 @@ if __name__ == "__main__":
                         })
         
     print('------------ Saving results ---------------')
-    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
-    
-    # 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")
+    with open("submission.json", "w") as f:
+        json.dump(solution, f)
 
     print("------------ Done ------------ ")