From 95ed452feb96236c1d6ba0e7c824f1accd1f9447 Mon Sep 17 00:00:00 2001 From: Brian Wylie Date: Tue, 10 Dec 2024 12:05:19 -0700 Subject: [PATCH] only checking the prediction column for NaN when we get back results from the endpoint --- src/sageworks/core/artifacts/endpoint_core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sageworks/core/artifacts/endpoint_core.py b/src/sageworks/core/artifacts/endpoint_core.py index 1fbb49126..c7a02a720 100644 --- a/src/sageworks/core/artifacts/endpoint_core.py +++ b/src/sageworks/core/artifacts/endpoint_core.py @@ -477,7 +477,7 @@ def _predict(self, eval_df: pd.DataFrame) -> pd.DataFrame: converted_df = converted_df.convert_dtypes() # Report on any rows that failed - failed_rows = converted_df[converted_df.isna().any(axis=1)] + failed_rows = converted_df[converted_df["prediction"].isna()] if not failed_rows.empty: self.log.warning(f"Rows that failed:\n{failed_rows}")