Hmm please see the attached. I have to put the "init" file under the

src

folder and change the import to

src.{project_name}.extras.datasets.{custom_dataset_file}

to make it work in ipython though.