Update script.py

script.py

@@ -8,13 +8,15 @@
 
 '''---compulsory---'''
 import hoho; hoho.setup() # YOU MUST CALL hoho.setup() BEFORE ANYTHING ELSE
+'''---compulsory---'''
+
 from pathlib import Path
 from tqdm import tqdm
 import pandas as pd
 import numpy as np
 
 
-def empty_solution():
+def empty_solution(sample):
     '''Return a minimal valid solution, i.e. 2 vertices and 1 edge.'''
     return np.zeros((2,3)), [(0, 1)]
 
@@ -26,7 +28,9 @@ if __name__ == "__main__":
     print('------------ Now you can do your solution ---------------')
     solution = []
     for i, sample in enumerate(tqdm(dataset)):
-        pred_vertices, pred_edges = empty_solution()
+        # replace this with your solution
+        pred_vertices, pred_edges = empty_solution(sample)
+        
         solution.append({
                         '__key__': sample['__key__'], 
                         'wf_vertices': pred_vertices.tolist(),