gene_tokenizer/__init__.py

from .configuration_nicheformer import NicheformerConfig
from .modeling_nicheformer import (
    NicheformerPreTrainedModel,
    NicheformerModel,
    NicheformerForMaskedLM
)