diff --git a/deepchopper/__init__.py b/deepchopper/__init__.py index f4854fe..0ac4849 100644 --- a/deepchopper/__init__.py +++ b/deepchopper/__init__.py @@ -1,8 +1,7 @@ """DeepChopper package.""" -from . import data, eval, models, train, ui, utils -from .cli import app +from . import cli, data, eval, models, train, ui, utils from .deepchopper import * # noqa: F403 from .models import DeepChopper -__all__ = ["models", "utils", "data", "train", "eval", "DeepChopper", "ui", "app"] +__all__ = ["models", "utils", "data", "train", "eval", "DeepChopper", "ui", "cli"] diff --git a/pyproject.toml b/pyproject.toml index 963a6e0..42a7516 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -62,7 +62,7 @@ deepchopper = "deepchopper.cli:app" [tool.poetry.scripts] dc-train = "deepchopper.train:main" dc-eval = "deepchopper.eval:main" -deepchopper = "deepchopper:app" +deepchopper = "deepchopper.cli:app" [tool.poe.tasks.train] cmd = "dc-train"