AttributeError: module 'kale.ml' has no attribute 'Dataset

[rafa-aptitude]

In the Blue Book for Bulldozers example, the cell that is supposed to create the kale dataset throws that error. You can check the example here:
https://github.com/kubeflow-kale/kale/blob/kubecon21eu/examples/bulldozers-kaggle-competition/blue-book-bulldozers.ipynb

I run it after setting up the cluster in AWS. The rest of the notebook up to that point runs smoothly.

[gparrella12]

You can import the dataset with from kale.common.artifacts import Dataset. After, in task definition, similarly you can import task with from kale.types import MLTask and replace kale_ml.Task.REGRESSION with MLTask.SIMPLE_REGRESSION.