Skip to content

Commit e52e9ac

Browse files
committed
fixed nan edge-case in report api
1 parent 1b2d56d commit e52e9ac

File tree

1 file changed

+1
-0
lines changed
  • giskard_vision/landmark_detection/tests

1 file changed

+1
-0
lines changed

giskard_vision/landmark_detection/tests/report.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ def to_dataframe(self, summary: Optional[bool] = False):
6969
raise GiskardImportError(["pandas"]) from e
7070

7171
df = pd.DataFrame(self.results)
72+
df.dropna(inplace=True) # in case the model failed to predict for instance
7273

7374
for col in ["metric_value", "prediction_time", "prediction_fail_rate"]:
7475
col_name = f"Best({col})"

0 commit comments

Comments
 (0)