mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-06 19:47:44 +00:00
feat: add a check for whether values in score_df are NaN (#756)
* add a check for whether values in score_df are NaN * fix ci * change raise to assert
This commit is contained in:
@@ -128,5 +128,10 @@ assert model_set_in_scores == set({{model_names}}).union({"ensemble"}), (
|
||||
assert score_df.index.is_unique, "The scores dataframe has duplicate model names."
|
||||
assert score_df.columns.tolist() == ["{{metric_name}}"], f"The column names of the scores dataframe should be ['{{metric_name}}'], but is '{score_df.columns.tolist()}'"
|
||||
|
||||
# Check for NaN values in score_df
|
||||
assert not score_df.isnull().values.any(), (
|
||||
f"The scores dataframe contains NaN values at the following locations:\n{score_df[score_df.isnull().any(axis=1)]}"
|
||||
)
|
||||
|
||||
|
||||
print("Ensemble test end.")
|
||||
|
||||
Reference in New Issue
Block a user