Skip to content

Commit

Permalink
refactor: Update import statement for cli in __init__.py
Browse files Browse the repository at this point in the history
  • Loading branch information
cauliyang committed Oct 11, 2024
1 parent 1e18479 commit 923e217
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
5 changes: 2 additions & 3 deletions deepchopper/__init__.py
Original file line number Diff line number Diff line change
@@ -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"]
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit 923e217

Please sign in to comment.