-
Notifications
You must be signed in to change notification settings - Fork 2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Skip files that don't have enough points to create a neighbourhood #109
Conversation
1992417
to
6433aa1
Compare
bird_cloud_gnn/radar_dataset.py
Outdated
@@ -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" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe it is worth adding the origin
to the message as it makes it easier to identify which file this occurs
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Added origin
Should we have a tests of this? |
6433aa1
to
7e2f87a
Compare
Hi @bart1, thanks for the review. I've added the origin and a test. I also rebased. |
7e2f87a
to
b5ac1e4
Compare
b5ac1e4
to
c9a621d
Compare
Kudos, SonarCloud Quality Gate passed! |
Description
If the data does not have enough points, skip it.
Related issues:
Closes #108
Instructions to review the pull request
The CI tests are enough