Skip to content

Commit 33d78cd

Browse files
committed
refactor: replace ValueError with ValidationError for calibration class validation
1 parent aec2010 commit 33d78cd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/mavedb/lib/validation/dataframe/calibration.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,9 @@ def validate_and_standardize_calibration_classes_dataframe(
5050
and data content.
5151
"""
5252
if not calibration.class_based:
53-
raise ValueError("Calibration classes file can only be provided for functional classification calibrations.")
53+
raise ValidationError(
54+
"Calibration classes file can only be provided for functional classification calibrations."
55+
)
5456

5557
standardized_classes_df = standardize_dataframe(classes_df, STANDARD_CALIBRATION_COLUMNS)
5658
validate_calibration_df_column_names(standardized_classes_df)
@@ -174,7 +176,7 @@ def validate_calibration_classes(
174176
that are missing from the series.
175177
"""
176178
if not calibration.functional_classifications:
177-
raise ValueError("Calibration must have functional classifications defined for class validation.")
179+
raise ValidationError("Calibration must have functional classifications defined for class validation.")
178180

179181
defined_classes = {c.class_ for c in calibration.functional_classifications}
180182
provided_classes = set(classes.tolist())

0 commit comments

Comments
 (0)