ModuleNotFoundError: No module named 'capslayer.data.datasets.stanford_drone'

[akorez]

Hi, I created the datareader and datawriter according to my own dataset by following the tutorials.
I made the necessary changes to config.py and main.py files. But I get an error as mentioned in the title.
How can I create my own module similar to the cifar10 module? Please, help me! Thanks...

[naturomics]

python can not find your dataloader, you should import your module to this init.py file

[akorez]

I added my own module into the init.py file. Here's the code. I still get the same error.

from future import absolute_import
from future import division
from future import print_function

from . import mnist
from . import fashion_mnist
from . import cifar10
from . import cifar100
from . import stanford_drone

all = ['mnist', 'fashion_mnist', 'cifar10', 'cifar100','stanford_drone']