add exception for Nonetype

2019-07-01 16:26:53 +02:00 · 2019-07-01 16:26:53 +02:00 · f30fd24b2b
commit f30fd24b2b
parent b0c3f51897
1 changed files with 1 additions and 1 deletions
--- a/src/predict/factory.py
+++ b/src/predict/factory.py
@ -15,7 +15,7 @@ def factory_for_gt(im_size, name=None, path_gt=None):
        with open(path_gt, 'r') as f:
            dic_names = json.load(f)
        print('-' * 120 + "\nMonoloco: Ground-truth file opened")
-    except FileNotFoundError:
+    except (FileNotFoundError, TypeError):
        print('-' * 120 + "\nMonoloco: ground-truth file not found\n")
        dic_names = {}