fix: make sure to include new implementation in bundle build

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
Aaron Pham
2024-03-20 22:11:53 +00:00
parent 5c8c30a70b
commit f0ab6d44fa
2 changed files with 5 additions and 4 deletions

View File

@@ -144,7 +144,7 @@ root = ".."
[tool.hatch.metadata]
allow-direct-references = true
[tool.hatch.build.targets.wheel]
only-include = ["src/openllm", "src/openllm_cli"]
only-include = ["src/openllm", "src/openllm_cli", "src/_openllm_tiny"]
sources = ["src"]
[tool.hatch.build.targets.sdist]
exclude = ["/.git_archival.txt", "tests", "/.python-version-default"]
@@ -163,7 +163,7 @@ dependencies = [
"types-protobuf",
]
enable-by-default = false
exclude = ["src/openllm/_service.py", "src/openllm/utils/__init__.py"]
exclude = ["src/_openllm_tiny/_service.py", "src/openllm/utils/__init__.py"]
include = [
"src/openllm/__init__.py",
"src/openllm/_quantisation.py",