Skip to content

Commit a07ec9d

Browse files
author
Pablo
committed
fix: better handler null url error cases
1 parent 972f756 commit a07ec9d

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

sdk/diffgram/core/diffgram_dataset_iterator.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,10 +120,14 @@ def __validate_file_ids(self):
120120

121121
def get_image_data(self, diffgram_file):
122122
MAX_RETRIES = 10
123+
image = None
123124
if hasattr(diffgram_file, 'image'):
124125
for i in range(0, MAX_RETRIES):
125126
try:
126-
image = imread(diffgram_file.image.get('url_signed'))
127+
if diffgram_file.image:
128+
url = diffgram_file.image.get('url_signed')
129+
if url:
130+
image = imread(diffgram_file.image.get('url_signed'))
127131
break
128132
except Exception as e:
129133
if i < MAX_RETRIES - 1:
@@ -163,7 +167,6 @@ def gen_tag_instances(self, instance_list):
163167
def get_file_instances(self, diffgram_file):
164168
if diffgram_file.type not in ['image', 'frame']:
165169
raise NotImplementedError('File type "{}" is not supported yet'.format(diffgram_file['type']))
166-
167170
image = self.get_image_data(diffgram_file)
168171
instance_list = diffgram_file.instance_list
169172
instance_types_in_file = set([x['type'] for x in instance_list])

0 commit comments

Comments
 (0)