chesstransformer/datasets/__init__.py

from .lichess import LichessDataset
from .lichess_simple_uci import LichessSimpleUciDataset
from .puzzle_dataset import LichessPuzzleDataset, LichessPuzzleFullSolutionDataset

__all__ = [
    "LichessDataset",
    "LichessSimpleUciDataset",
    "LichessPuzzleDataset",
    "LichessPuzzleFullSolutionDataset",
]