[demorunner] bugfix for exception handling during demo runtime

ipol-journal · Jun 26, 2023 · 1a2262f · 1a2262f
1 parent 7a38657
commit 1a2262f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/ipol_demo/modules/core/core.py b/ipol_demo/modules/core/core.py
@@ -1744,7 +1744,7 @@ def execute_experiment(
             self.logger.exception(error_message)
             raise IPOLExecutionError(error_message, error_message)
 
-        if status_code != 200:
+        if dr_response.get("error"):
             print("DR answered KO for demo #{}".format(demo_id))
 
             try:

diff --git a/ipol_demo/modules/demorunner/demorunner.py b/ipol_demo/modules/demorunner/demorunner.py
@@ -45,9 +45,8 @@
     status,
 )
 from fastapi.exceptions import RequestValidationError
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, StreamingResponse
 from pydantic import BaseSettings
-from starlette.responses import StreamingResponse
 from Tools.run_demo_base import IPOLTimeoutError