KeyError when downloading a dataset with missing segmentations

[stevensagaert]

Here's the stack trace:

Traceback (most recent call last):
File "/home/steven.sagaert/python_workspace/FARAD2SORT/FARAD_research/Steven/Active/segments_interface.py", line 94, in
dataset = fetch_release(dataset_name=dataset_name)
File "/home/steven.sagaert/python_workspace/FARAD2SORT/FARAD_research/Steven/Active/segments_interface.py", line 28, in fetch_release
return SegmentsDataset(release,filter_by="REVIEWED", segments_dir=dump_folder)
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/site-packages/segments/dataset.py", line 155, in init
self.load_dataset()
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/site-packages/segments/dataset.py", line 214, in load_dataset
list(
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/site-packages/tqdm/std.py", line 1178, in iter
for obj in iterable:
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/multiprocessing/pool.py", line 868, in next
raise value
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/multiprocessing/pool.py", line 125, in worker
result = (True, func(*args, **kwds))
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/site-packages/segments/dataset.py", line 198, in _load_image
self.getitem(i)
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/site-packages/segments/dataset.py", line 325, in getitem
segmentation_bitmap = self._load_segmentation_bitmap_from_cache(
File "/home/steven.sagaert/miniconda3/envs/farad/lib/python3.8/site-packages/segments/dataset.py", line 259, in _load_segmentation_bitmap_from_cache
segmentation_bitmap_url = label["attributes"]["segmentation_bitmap"]["url"]
KeyError: 'segmentation_bitmap'

The solution is to change line 337 in dataset.py to
except (KeyError,TypeError):

[dbbert]

Thanks for the bug report, we'll include this fix in the next SDK release. @segments-arnaud FYI