Exception fix (#236)

Exception Handling while loading models. Signed-off-by: kbatra <[email protected]>
numaproj · Jul 28, 2023 · e3f90db · e3f90db
1 parent 5a0ef08
commit e3f90db
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/udf/anomaly-detection/src/udf/inference.py b/udf/anomaly-detection/src/udf/inference.py
@@ -93,6 +93,16 @@ def inference(
             )
             return static_response
 
+        except Exception as ex:
+            _LOGGER.exception(
+                "%s - Unhandled exception while fetching  inference artifact, Keys: %s, Metric: %s, Error: %r",
+                payload.uuid,
+                payload.composite_keys,
+                payload.metrics,
+                ex,
+            )
+            return static_response
+
         # Check if artifact is found
         if not artifact_data:
             _LOGGER.info(

diff --git a/udf/anomaly-detection/src/udf/preprocess.py b/udf/anomaly-detection/src/udf/preprocess.py
@@ -67,6 +67,16 @@ def preprocess(self, keys: List[str], payload: StreamPayload) -> (np.ndarray, St
             )
             return None, Status.RUNTIME_ERROR
 
+        except Exception as ex:
+            _LOGGER.exception(
+                "%s - Unhandled exception while fetching preproc artifact, Keys: %s, Metric: %s, Error: %r",
+                payload.uuid,
+                payload.composite_keys,
+                payload.metrics,
+                ex,
+            )
+            return None, Status.RUNTIME_ERROR
+
         # Check if artifact is found
         if not preproc_artifact:
             _LOGGER.info(

diff --git a/udf/anomaly-detection/src/udf/threshold.py b/udf/anomaly-detection/src/udf/threshold.py
@@ -64,6 +64,17 @@ def threshold(
             static_scores = calculate_static_thresh(metric_arr, static_thresh)
             return static_scores, Status.RUNTIME_ERROR, Header.STATIC_INFERENCE, -1
 
+        except Exception as ex:
+            _LOGGER.exception(
+                "%s - Unhandled exception while fetching threshold artifact, Keys: %s, Metric: %s, Error: %r",
+                payload.uuid,
+                payload.composite_keys,
+                payload.metrics,
+                ex,
+            )
+            static_scores = calculate_static_thresh(metric_arr, static_thresh)
+            return static_scores, Status.RUNTIME_ERROR, Header.STATIC_INFERENCE, -1
+
         # Check if artifact is found
         if not thresh_artifact:
             _LOGGER.info(