Skip to content

Commit

Permalink
FIX: refer to arg model (not pipeline) in LearnerInspector exceptions (
Browse files Browse the repository at this point in the history
  • Loading branch information
j-ittner authored Jul 5, 2023
1 parent b389393 commit 6749064
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/facet/inspection/_learner_inspector.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ def __init__(
"""

if not model.is_fitted:
raise ValueError("arg pipeline must be fitted")
raise ValueError("arg model must be fitted")

final_estimator: T_SupervisedLearnerDF = model.final_estimator
if is_classifier(final_estimator):
Expand All @@ -119,7 +119,7 @@ def __init__(
)
elif not is_regressor(final_estimator):
raise TypeError(
"learner in arg pipeline must be a classifier or a regressor,"
"learner in arg model must be a classifier or a regressor,"
f"but is a {type(final_estimator).__name__}"
)

Expand Down

0 comments on commit 6749064

Please sign in to comment.