Skip to content

Commit

Permalink
Added useful error message
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexejPenner committed Jul 2, 2024
1 parent 04ab578 commit 87b7e10
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions template/steps/training/model_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,10 @@

experiment_tracker = Client().active_stack.experiment_tracker

if not experiment_tracker:
raise RuntimeError("Please add an experiment tracker to your stack in order to run the pipeline"
"containing the `model_evaluator` step.")


@step(experiment_tracker=experiment_tracker.name)
def model_evaluator(
Expand Down

0 comments on commit 87b7e10

Please sign in to comment.