model/basenji/__init__.py

"""Minimal Basenji modules required by Saluki."""

import os


os.environ.setdefault("TF_USE_LEGACY_KERAS", "1")

from .dataset import RnaDataset
from .rnann import RnaNN
from .trainer import Trainer

__all__ = ["RnaDataset", "RnaNN", "Trainer"]