only checking the prediction column for NaN when we get back results …

…from the endpoint

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}")