from .ngboost import NGBoost
from .api import NGBClassifier, NGBRegressor, NGBSurvival

__version__ = "0.2.1dev"

from .ngboost import NGBoost
from .api import NGBClassifier, NGBRegressor, NGBSurvival

Update __init__.py