[FIX] Catch unknown assignment error

into-ai · Mar 27, 2020 · b9702bf · b9702bf
1 parent f5b28d7
commit b9702bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kerasltiprovider/provider.py b/kerasltiprovider/provider.py
@@ -68,7 +68,7 @@ def inputs(assignment_id: AnyIDType) -> RequestResultType:
 
         try:
             assignment = find_assignment(assignment_id)
-        except (TypeError, ValueError, IndexError):
+        except UnknownAssignmentException:
             log.warning(
                 f"Ingoring request of validation inputs for unknown assignment: {assignment_id}"
             )
@@ -115,7 +115,7 @@ def inputs_single(assignment_id: AnyIDType, input_id: int) -> RequestResultType:
 
         try:
             assignment = find_assignment(assignment_id)
-        except (TypeError, ValueError, IndexError):
+        except UnknownAssignmentException:
             log.warning(
                 f"Ingoring request of validation inputs for unknown assignment: {assignment_id}"
             )