Skip files that don't have enough points to create a neighbourhood

bird_cloud_gnn/radar_dataset.py

@@ -116,6 +116,11 @@ def _process_data(self, data, origin=""):
                 )
             ].index
         ).reset_index(drop=True)
+        if len(data) < self.num_neighbours:
+            print(
+                f"Warning: There are not enough points to find {self.num_neighbours} neighbourhood"
+            )
+            return
 
         data_xyz = data[xyz]
         # remove the special features so they can be generated later