usm3d
/

empty_submission

Model card Files Files and versions Community

dmytromishkin commited on Mar 15

Commit

7c466d7

•

1 Parent(s): 299c8f2

Update script.py

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -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.to_parquet(Path(params['output_path']) / "submission.parquet")
     print("------------ Done ------------ ")

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