Skip to content

Commit 6ad1e26

Browse files
authored
Merge pull request #118 from georgian-io/script-entrypoint
Modify Script Entrypoint to be `llmtune` instead of `python toolkit.py`
2 parents e958635 + 404bff6 commit 6ad1e26

File tree

5 files changed

+6
-4
lines changed

5 files changed

+6
-4
lines changed

Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ build-release:
22
rm -rf dist
33
rm -rf build
44
poetry build
5-
poetry publis
5+
poetry publish

llmtune/cli/__init__.py

Whitespace-only changes.

toolkit.py renamed to llmtune/cli/toolkit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,5 +109,5 @@ def run(config_path: str = "./config.yml") -> None:
109109
run_one_experiment(config, config_path)
110110

111111

112-
if __name__ == "__main__":
112+
def cli():
113113
app()

poetry.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@ license = "Apache 2.0"
77
readme = "README.md"
88
packages = [{include = "llmtune"}]
99

10+
[tool.poetry.scripts]
11+
llmtune = "llmtune.cli.toolkit:cli"
12+
1013
[tool.poetry.dependencies]
1114
python = ">=3.9, <=3.12"
1215
transformers = "~4.37.2"
@@ -43,7 +46,6 @@ shellingham = "^1.5.4"
4346

4447

4548
[tool.poetry.group.dev.dependencies]
46-
twine = "^5.0.0"
4749
ruff = "~0.3.5"
4850

4951
[build-system]

0 commit comments

Comments
 (0)