remove edge semantics

handcrafted_solution.py

@@ -240,4 +240,4 @@ def predict(entry, visualize=False) -> Tuple[np.ndarray, List[int]]:
         from hoho.viz3d import plot_estimate_and_gt
         plot_estimate_and_gt(all_3d_vertices_clean, connections_3d_clean, good_entry['wf_vertices'],
                             good_entry['wf_edges'])
-    return good_entry['__key__'], all_3d_vertices_clean, connections_3d_clean, [0 for i in range(len(connections_3d_clean))]
+    return good_entry['__key__'], all_3d_vertices_clean, connections_3d_clean

script.py

@@ -134,8 +134,7 @@ if __name__ == "__main__":
             solution.append({
                             '__key__': key,
                             'wf_vertices': pred_vertices.tolist(),
-                            'wf_edges': pred_edges,
-                            'edge_semantics': semantics,
+                            'wf_edges': pred_edges
                         })
             if i % 100 == 0:
                 # incrementally save the results in case we run out of time