kubeflow-kale / kale

Kubeflow’s superfood for Data Scientists
http://kubeflow-kale.github.io
Apache License 2.0
632 stars 128 forks source link

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

Open rafa-aptitude opened 2 years ago

rafa-aptitude commented 2 years ago

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 commented 1 year ago

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.