Skip to content

Commit

Permalink
Check the list of tasks against
Browse files Browse the repository at this point in the history
Thinking about it, probably the smarter thing to do would be to simply
not specify it in parameters_combinations but just use the keys from
tasks_to_thresholds. But I don't want to attempt that right now, since
the workflow is working now.
  • Loading branch information
spigo900 committed Aug 11, 2020
1 parent d2cdf5d commit 4a42d92
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions ai2/pegasus.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,10 @@ def main(params: Parameters):
ensemble_params = ensemble_params.unify({'task_to_gold': {
task: {'val_y': task_params.existing_file('val_y')}
}})
if not task_to_jobs_info.get(task):
raise RuntimeError(
"Each task with a threshold must appear in the list of task parameters."
)

ensemble_params = ensemble_params.unify({
'data_sizes': data_sizes,
Expand Down

0 comments on commit 4a42d92

Please sign in to comment.