shape (n_tasks,)

same below. Remove the "=" before shape