ModuleNotFoundError: No module named 'dataset'

[leeuack]

Hi,
after reading your paper, I am trying to reproduce your works.
Firstly, I trained VAE with the chair dataset.
Before testing tsne space, I am testing plotting process.
When I run "python create_plot.py color-test", it occurs ModuleNotFoundError.

In the code (create_plot.py),

from dataset import dataset as dataset
dataset.load_voxels('cpu')
dataset.load_labels()
voxels = dataset.voxels

but I could not find 'dataset'.
is it a typo of datasets?
Also, I could not find where load_voxels() and load_labels() are defined.
could you clarify which dataset it is pointing to?

Thank you,

[lz-503]

Hi，I also encounter this problem. How did you solve it？ I am looking forward to your reply. Thank you very much！