Hugging Face's logo

usm3d
/
empty_submission

Model card Files Community
1
dmytromishkin commited on
Commit
35f02b8
·
verified ·
1 Parent(s): b4bc845

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +1 -2
script.py CHANGED
@@ -30,8 +30,7 @@ if __name__ == "__main__":
30
  solution.append({
31
  '__key__': sample['__key__'],
32
  'wf_vertices': pred_vertices.tolist(),
33
- 'wf_edges': pred_edges,
34
- 'edge_semantics': semantics,
35
  })
36
  print('------------ Saving results ---------------')
37
  sub = pd.DataFrame(solution, columns=["__key__", "wf_vertices", "wf_edges", "edge_semantics"])
 
30
  solution.append({
31
  '__key__': sample['__key__'],
32
  'wf_vertices': pred_vertices.tolist(),
33
+ 'wf_edges': pred_edges
 
34
  })
35
  print('------------ Saving results ---------------')
36
  sub = pd.DataFrame(solution, columns=["__key__", "wf_vertices", "wf_edges", "edge_semantics"])