fixed bug when passing filename that doesn't exist into heat_map_analysis
@@ -1144,6 +1144,15 @@ class Model(object):
elif filename:
temp = data[data['_filename_0'].isin(filename)]
image_id = temp['_id_'].tolist()
+ if not image_id:
+ raise ValueError('filename: {} not found in table'.format(filename))
+
+
+ # filter images by id number
+ if image_id:
+ data = data[data['_id_'].isin(image_id)]
+ if data.numrows().numrows == 0:
+ raise ValueError('image_id: {} not found in the table'.format(image_id))
# filter images by id number
