Update script.py

script.py

@@ -33,6 +33,6 @@ if __name__ == "__main__":
                         'wf_edges': pred_edges
                     })
     print('------------ Saving results ---------------')
-    sub = pd.DataFrame(solution, columns=["__key__", "wf_vertices", "wf_edges", "edge_semantics"])
+    sub = pd.DataFrame(solution, columns=["__key__", "wf_vertices", "wf_edges"])
     sub.to_parquet(Path(params['output_path']) / "submission.parquet")
     print("------------ Done ------------ ")