add try and except to um.py

UM-Bridge · Nov 13, 2024 · b50118b · b50118b
1 parent 660d146
commit b50118b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/umbridge/um.py b/umbridge/um.py
@@ -206,8 +206,11 @@ async def evaluate(request):
             if len(parameters[i]) != input_sizes[i]:
                 return error_response("InvalidInput", f"Input parameter {i} has invalid length! Expected {input_sizes[i]} but got {len(parameters[i])}.", 400)
 
-        output_future = model_executor.submit(model.__call__, parameters, config)
-        output = await asyncio.wrap_future(output_future)
+        try:
+            output_future = model_executor.submit(model.__call__, parameters, config)
+            output = await asyncio.wrap_future(output_future)
+        except Exception as e:
+            print(f"An error occured during the evaluation: {e}")
 
         # Check if output is a list of lists
         if not isinstance(output, list):