Skip to content

Commit

Permalink
[shark-ai] Include sharktank in the meta package (#701)
Browse files Browse the repository at this point in the history
  • Loading branch information
marbre authored Dec 16, 2024
1 parent d1980c7 commit b151ffa
Showing 1 changed file with 11 additions and 13 deletions.
24 changes: 11 additions & 13 deletions build_tools/python_deploy/write_requirements.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,8 @@ def write_requirements(requirements):
metapackage_version = load_version_info(VERSION_FILE_LOCAL)
PACKAGE_VERSION = metapackage_version.get("package-version")

# sharktank_version = load_version_info(VERSION_FILE_SHARKTANK)
# SHARKTANK_PACKAGE_VERSION = sharktank_version.get("package-version")
sharktank_version = load_version_info(VERSION_FILE_SHARKTANK)
SHARKTANK_PACKAGE_VERSION = sharktank_version.get("package-version")

shortfin_version = load_version_info(VERSION_FILE_SHORTFIN)
SHORTFIN_PACKAGE_VERSION = shortfin_version.get("package-version")
Expand All @@ -65,13 +65,12 @@ def write_requirements(requirements):
requirements = ""
for package in stable_packages_list:
requirements += package + "\n"
# TODO: Include sharktank as a dependencies of future releases
# requirements = (
# "sharktank=="
# + Version(SHARKTANK_PACKAGE_VERSION).base_version
# + args.version_suffix
# + "\n"
# )
requirements = (
"sharktank=="
+ Version(SHARKTANK_PACKAGE_VERSION).base_version
+ args.version_suffix
+ "\n"
)
requirements += (
"shortfin=="
+ Version(SHORTFIN_PACKAGE_VERSION).base_version
Expand All @@ -89,10 +88,9 @@ def write_requirements(requirements):
requirements = ""
for package in stable_packages_list:
requirements += package + "==" + STABLE_VERSION_TO_PIN + "\n"
# TODO: Include sharktank as a dependencies of future releases
# requirements += (
# "sharktank==" + Version(SHARKTANK_PACKAGE_VERSION).base_version + "\n"
# )
requirements += (
"sharktank==" + Version(SHARKTANK_PACKAGE_VERSION).base_version + "\n"
)
requirements += "shortfin==" + Version(SHORTFIN_PACKAGE_VERSION).base_version

write_requirements(requirements)

0 comments on commit b151ffa

Please sign in to comment.