Description:
In collect.py, the continue statement conditioned on annotation existence prevents SDMap from being added to the test set.
This causes the test split to miss SDMap entries even when they are available.
Proposed Fix:
Move the block of code responsible for adding SDMap before the annotation existence check. This ensures that SDMap is added to the test set regardless of whether annotations are present.