Skip to content

Commit b151ffa

Browse files
authored
[shark-ai] Include sharktank in the meta package (#701)
1 parent d1980c7 commit b151ffa

File tree

1 file changed

+11
-13
lines changed

1 file changed

+11
-13
lines changed

build_tools/python_deploy/write_requirements.py

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ def write_requirements(requirements):
5353
metapackage_version = load_version_info(VERSION_FILE_LOCAL)
5454
PACKAGE_VERSION = metapackage_version.get("package-version")
5555

56-
# sharktank_version = load_version_info(VERSION_FILE_SHARKTANK)
57-
# SHARKTANK_PACKAGE_VERSION = sharktank_version.get("package-version")
56+
sharktank_version = load_version_info(VERSION_FILE_SHARKTANK)
57+
SHARKTANK_PACKAGE_VERSION = sharktank_version.get("package-version")
5858

5959
shortfin_version = load_version_info(VERSION_FILE_SHORTFIN)
6060
SHORTFIN_PACKAGE_VERSION = shortfin_version.get("package-version")
@@ -65,13 +65,12 @@ def write_requirements(requirements):
6565
requirements = ""
6666
for package in stable_packages_list:
6767
requirements += package + "\n"
68-
# TODO: Include sharktank as a dependencies of future releases
69-
# requirements = (
70-
# "sharktank=="
71-
# + Version(SHARKTANK_PACKAGE_VERSION).base_version
72-
# + args.version_suffix
73-
# + "\n"
74-
# )
68+
requirements = (
69+
"sharktank=="
70+
+ Version(SHARKTANK_PACKAGE_VERSION).base_version
71+
+ args.version_suffix
72+
+ "\n"
73+
)
7574
requirements += (
7675
"shortfin=="
7776
+ Version(SHORTFIN_PACKAGE_VERSION).base_version
@@ -89,10 +88,9 @@ def write_requirements(requirements):
8988
requirements = ""
9089
for package in stable_packages_list:
9190
requirements += package + "==" + STABLE_VERSION_TO_PIN + "\n"
92-
# TODO: Include sharktank as a dependencies of future releases
93-
# requirements += (
94-
# "sharktank==" + Version(SHARKTANK_PACKAGE_VERSION).base_version + "\n"
95-
# )
91+
requirements += (
92+
"sharktank==" + Version(SHARKTANK_PACKAGE_VERSION).base_version + "\n"
93+
)
9694
requirements += "shortfin==" + Version(SHORTFIN_PACKAGE_VERSION).base_version
9795

9896
write_requirements(requirements)

0 commit comments

Comments
 (0)