Update script.py

script.py

@@ -16,7 +16,7 @@ import numpy as np
 
 def empty_solution():
     '''Return a minimal valid solution, i.e. 2 vertices and 1 edge.'''
-    return np.zeros((2,3)), [(0, 1)], [0]
+    return np.zeros((2,3)), [(0, 1)]
 
 
 if __name__ == "__main__":
@@ -26,7 +26,7 @@ if __name__ == "__main__":
     print('------------ Now you can do your solution ---------------')
     solution = []
     for i, sample in enumerate(tqdm(dataset)):
-        pred_vertices, pred_edges, semantics = empty_solution()
+        pred_vertices, pred_edges = empty_solution()
         solution.append({
                         '__key__': sample['__key__'], 
                         'wf_vertices': pred_vertices.tolist(),