──────────────────── Failures during collection ─────────────────────
─ Could not collect xxxxxxxxxxxxxxxxxxxxxxx… ─
ValueError: Task 'task_xxxxxxxxxxxxx' is parametrized with
'arg_names' ('depends_on', 'model', 'sample_kwargs', 'produces') with
4 elements, but the number of provided 'arg_values' is 3.