Update script.py

script.py

@@ -121,6 +121,7 @@ if __name__ == "__main__":
             except Exception as e:
                 print (f"Faile due to {e}")
                 pred_vertices, pred_edges = empty_solution(sample)
+            print (f'{pred_vertices=}, {pred_edges=}')
             solution.append({
                             'order_id': sample['order_id'], 
                             'wf_vertices': pred_vertices.tolist(),
@@ -128,6 +129,9 @@ if __name__ == "__main__":
                         })
         
     print('------------ Saving results ---------------')
+    print (len(solution))
+    from time import sleep
+    sleep(30)
     sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
     sub.to_parquet("submission.parquet")
     print("------------ Done ------------ ")