diff --git a/python-sdk/nuscenes/nuscenes.py b/python-sdk/nuscenes/nuscenes.py index c36c023ce..f8ca893bb 100644 --- a/python-sdk/nuscenes/nuscenes.py +++ b/python-sdk/nuscenes/nuscenes.py @@ -794,10 +794,11 @@ def list_scenes(self) -> None: def ann_count(record): count = 0 - sample = self.nusc.get('sample', record['first_sample_token']) - while not sample['next'] == "": + next_sample_token = record['first_sample_token'] + while next_sample_token: + sample = self.nusc.get('sample', next_sample_token) count += len(sample['anns']) - sample = self.nusc.get('sample', sample['next']) + next_sample_token = sample["next"] return count recs = [(self.nusc.get('sample', record['first_sample_token'])['timestamp'], record) for record in