Update script.py

script.py

@@ -15,11 +15,10 @@ from joblib import Parallel, delayed
 import os
 import json
 import gc
-
 from hoho2025.example_solutions import predict_wireframe
 # check the https://github.com/s23dr/hoho2025/blob/main/hoho2025/example_solutions.py for the example solution
 
-def empty_solution(sample):
+def empty_solution():
     '''Return a minimal valid solution, i.e. 2 vertices and 1 edge.'''
     return np.zeros((2,3)), [(0, 1)]
 
@@ -105,7 +104,7 @@ if __name__ == "__main__":
         try:
             pred_vertices, pred_edges = predict_wireframe(sample)
         except:
-            pred_vertices, pred_edges = empty_solution(sample)
+            pred_vertices, pred_edges = empty_solution()
         if i %10 == 0:
             gc.collect()
         return {
@@ -114,13 +113,12 @@ if __name__ == "__main__":
             'wf_edges': pred_edges
         }
     num_cores = 4
-    for subset_name in dataset:
-        samples = list(dataset[subset_name])
-        # Process samples in parallel with simple tqdm progress tracking
-        results = Parallel(n_jobs=num_cores)(
-            delayed(process_sample)(sample, i) for sample in tqdm(samples, desc=f"Processing {subset_name}")
-        )
-        solution.extend(results)
+    
+    for subset_name in dataset.keys():
+        print (f"Predicting {subset_name}")
+        for i, sample in enumerate(tqdm(dataset[subset_name])):
+            res = process_sample(sample, i)
+            solution.append(res)
         
     print('------------ Saving results ---------------')
     sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])