magma / magma /datasets /__init__.py
stellaathena's picture
This should work
bb5cd12
from .dataset import (
ImgCptDataset,
collate_fn,
)