return key with empty solution

handcrafted_solution.py

@@ -235,9 +235,9 @@ def predict(entry, visualize=False) -> Tuple[np.ndarray, List[int]]:
     all_3d_vertices_clean, connections_3d_clean  = prune_not_connected(all_3d_vertices, connections_3d)
     if (len(all_3d_vertices_clean) < 2) or len(connections_3d_clean) < 1:
         print (f'Not enough vertices or connections in the 3D vertices')
-        return empty_solution()
+        return (good_entry['__key__'], *empty_solution())
     if visualize:
-        from viz3d import plot_estimate_and_gt
+        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))]

script.py

@@ -102,6 +102,7 @@ def convert_entry_to_human_readable(entry):
 
 ### The part below is used to define and test your solution.
 
+from pathlib import Path
 def save_submission(submission, path):
     """
     Saves the submission to a specified path.
@@ -119,6 +120,7 @@ if __name__ == "__main__":
     print ("------------ Loading dataset------------ ")
     params = hoho.get_params()
     dataset = hoho.get_dataset(decode=None, split='all', dataset_type='webdataset')
+
     print('------------ Now you can do your solution ---------------')
     solution = []
     from concurrent.futures import ProcessPoolExecutor