diff --git a/compute_worker/compute_worker.py b/compute_worker/compute_worker.py index f9b74d27b..f42b1b6dc 100644 --- a/compute_worker/compute_worker.py +++ b/compute_worker/compute_worker.py @@ -376,6 +376,7 @@ def _get_container_image(self, image_name): docker_pull_fail_data = { "type": "Docker_Image_Pull_Fail", "error_message": error_message, + "is_scoring": self.is_scoring } # Send data to be written to ingestion logs self._update_submission(docker_pull_fail_data) diff --git a/src/apps/api/views/submissions.py b/src/apps/api/views/submissions.py index 6ce9da84b..b61963f9b 100644 --- a/src/apps/api/views/submissions.py +++ b/src/apps/api/views/submissions.py @@ -68,9 +68,8 @@ def check_object_permissions(self, request, obj): # Set file name to ingestion std error as default error_file_name = "prediction_ingestion_stderr" - # Change error file name when error comes from execution time limit - # and error occured during scoring - if request.data["type"] == "Execution_Time_Limit_Exceeded" and request.data['is_scoring'] == "True": + # Change error file name to scoring_stderr when error occurs during scoring + if request.data.get("is_scoring", "False") == "True": error_file_name = "scoring_stderr" try: