Skip to content

Commit

Permalink
Find torchserve-plugins-sdk artifacts locally (#2429)
Browse files Browse the repository at this point in the history
* Find torchserve-plugins-sdk artifacts locally

* Upgrade torchserve-plugins-sdk version

* Find torchserve-plugins-sdk locally for regression tests

* Add support to build and install trochserve-plugings-sdk in install_dependencies script

* revert modifications to workflow files

---------

Co-authored-by: Naman Nandan <namannan@amazon.com>
  • Loading branch information
namannandan and Naman Nandan authored Jun 23, 2023
1 parent 207477d commit a77a150
Show file tree
Hide file tree
Showing 4 changed files with 19 additions and 3 deletions.
1 change: 1 addition & 0 deletions frontend/build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ allprojects {
repositories {
mavenCentral()
jcenter()
mavenLocal()
}

apply plugin: 'idea'
Expand Down
4 changes: 2 additions & 2 deletions frontend/gradle.properties
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ netty_version=4.1.53.Final
slf4j_api_version=1.7.32
slf4j_log4j_version=2.17.1
testng_version=7.1.0
torchserve_sdk_version=0.0.4
torchserve_sdk_version=0.0.5
snakeyaml_version=1.31
grpc_version=1.50.0
protoc_version=3.18.0
lmax_disruptor_version=3.4.4
lmax_disruptor_version=3.4.4
2 changes: 1 addition & 1 deletion plugins/gradle.properties
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
gson_version=2.9.0
torchserve_sdk_version=0.0.4
torchserve_sdk_version=0.0.5
testng_version=7.1.0
15 changes: 15 additions & 0 deletions ts_scripts/install_dependencies.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,9 @@ def install_wget(self):
def install_numactl(self):
pass

def install_torchserve_plugins_sdk(self):
pass


class Linux(Common):
def __init__(self):
Expand Down Expand Up @@ -110,6 +113,9 @@ def install_numactl(self):
if os.system("numactl --show") != 0 or args.force:
os.system(f"{self.sudo_cmd}apt-get install -y numactl")

def install_torchserve_plugins_sdk(self):
os.system(f"cd {REPO_ROOT}/serving-sdk/ && ./mvnw clean install -q && cd ../")


class Windows(Common):
def __init__(self):
Expand All @@ -128,6 +134,11 @@ def install_wget(self):
def install_numactl(self):
pass

def install_torchserve_plugins_sdk(self):
os.system(
f"cd {REPO_ROOT}/serving-sdk/ && .\mvnw.cmd clean install -q && cd ../"
)


class Darwin(Common):
def __init__(self):
Expand Down Expand Up @@ -156,6 +167,9 @@ def install_numactl(self):
if os.system("numactl --show") != 0 or args.force:
os.system("brew install numactl")

def install_torchserve_plugins_sdk(self):
os.system(f"cd {REPO_ROOT}/serving-sdk/ && ./mvnw clean install -q && cd ../")


def install_dependencies(cuda_version=None, nightly=False):
os_map = {"Linux": Linux, "Windows": Windows, "Darwin": Darwin}
Expand All @@ -166,6 +180,7 @@ def install_dependencies(cuda_version=None, nightly=False):
system.install_nodejs()
system.install_node_packages()
system.install_numactl()
system.install_torchserve_plugins_sdk()

# Sequence of installation to be maintained
system.install_java()
Expand Down

0 comments on commit a77a150

Please sign in to comment.