hmm I have the "__init__.py" file under the

{project_name}/extras

folder. Then I specify the

type

of the model artifact in the

catalog.yml

file as

{project_name}.extras.datasets.{custom_dataset.py}.{custom_dataset_class}

. Anything else I missed?