Skip to content

Commit b40581f

Browse files
committed
[feat] Add script interface in pyproject.toml
1 parent d9df22c commit b40581f

File tree

1 file changed

+9
-19
lines changed

1 file changed

+9
-19
lines changed

pyproject.toml

Lines changed: 9 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ readme = 'README.md'
1111
requires-python = '>=3.10'
1212
license = 'MIT'
1313
keywords = []
14+
# FIXME: authors
1415
authors = [{ name = 'Chenhui Zhang', email = '[email protected]' }]
1516
classifiers = [
1617
'Development Status :: 4 - Beta',
@@ -26,7 +27,10 @@ dependencies = [
2627
"click~=8.1",
2728
"datasets~=3.2",
2829
"deepspeed~=0.16.3",
29-
"diffusers~=0.32.2",
30+
31+
# FIXME: change diffusers source later
32+
# "diffusers @ git+https://github.com/zRzRzRzRzRzRzR/diffusers.git@cogview4_control",
33+
3034
"imageio-ffmpeg~=0.6.0",
3135
"imageio~=2.37",
3236
"peft~=0.14.0",
@@ -41,6 +45,10 @@ dependencies = [
4145
torch = ["numpy", "torch", "torchvision"]
4246
video = ["decord~=0.6.0", "opencv-python-headless~=4.11"]
4347

48+
[project.scripts]
49+
# FIXME: change name to cogkit
50+
cogkit = "cogmodels.cli:cli"
51+
4452
[project.urls]
4553
# TODO: adds project urls
4654
# Documentation = 'https://github.com/Chenhui Zhang/cogmodels#readme'
@@ -87,21 +95,3 @@ ignore = [
8795
"UP009",
8896
"W293",
8997
]
90-
91-
[tool.uv.sources]
92-
diffusers = { git = "https://github.com/huggingface/diffusers" }
93-
94-
95-
# TODO: test coverage
96-
# [tool.coverage.run]
97-
# source_pkgs = ['cogmodels', 'tests']
98-
# branch = true
99-
# parallel = true
100-
# omit = ['src/cogmodels/__about__.py']
101-
102-
# [tool.coverage.paths]
103-
# cogmodels = ['*/cogmodels/src/cogmodels', 'src/cogmodels']
104-
# tests = ['*/cogmodels/tests', 'tests']
105-
106-
# [tool.coverage.report]
107-
# exclude_lines = ['if TYPE_CHECKING:', 'if __name__ == .__main__.:', 'no cov']

0 commit comments

Comments
 (0)