fixed nan edge-case in report api

Giskard-AI · Aug 19, 2024 · e52e9ac · e52e9ac
1 parent 1b2d56d
commit e52e9ac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/giskard_vision/landmark_detection/tests/report.py b/giskard_vision/landmark_detection/tests/report.py
@@ -69,6 +69,7 @@ def to_dataframe(self, summary: Optional[bool] = False):
             raise GiskardImportError(["pandas"]) from e
 
         df = pd.DataFrame(self.results)
+        df.dropna(inplace=True)  # in case the model failed to predict for instance
 
         for col in ["metric_value", "prediction_time", "prediction_fail_rate"]:
             col_name = f"Best({col})"