catch errors in load dataframe (#68)

Co-authored-by: Sam <[email protected]>
arcus-azure · Jun 15, 2020 · de20e8c · de20e8c
1 parent bbfda11
commit de20e8c
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/arcus/ml/images/io.py b/arcus/ml/images/io.py
@@ -147,6 +147,9 @@ def load_images_from_dataframe(df: pd.DataFrame, image_column_name:str, target_c
         except FileNotFoundError:
             _logger.warning('File ' + file + ' not found')
             filecheck_ok = False
+        except:
+            _logger.warning('File ' + file + ' resulted in an error')
+            filecheck_ok = False
 
         if filecheck_ok :
             images.append(im)

diff --git a/tests/resources/temp/io/__empty.py b/tests/resources/temp/io/__empty.py
diff --git a/tests/resources/temp/io/__emptyfile__ b/tests/resources/temp/io/__emptyfile__
@@ -0,0 +1 @@
+do not remove
diff --git a/tests/resources/temp/io/_empty.py b/tests/resources/temp/io/_empty.py