From 623db9e6c684cb06442fcc7ab35e565d13fd9367 Mon Sep 17 00:00:00 2001
From: schenqian <33588935+schenqian@users.noreply.github.com>
Date: Wed, 27 Jan 2021 12:19:47 -0800
Subject: [PATCH] fix: add required dependencies

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b704b3f..e4aa99b 100644
--- a/setup.py
+++ b/setup.py
@@ -48,7 +48,7 @@ def read(fname):
     ],
 
     # support sagemaker-inference==1.1.0 for mxnet 1.4 eia image and
-    install_requires=['sagemaker-inference>=1.1.0,<=1.5.2', 'retrying==1.3.3'],
+    install_requires=['model-archiver', 'multi-model-server', 'sagemaker-inference>=1.1.0,<=1.5.2', 'retrying==1.3.3'],
     extras_require={
         'test': ['tox', 'flake8', 'pytest', 'pytest-cov', 'pytest-xdist', 'pytest-rerunfailures',
                  'mock', 'sagemaker==1.62.0', 'docker-compose', 'mxnet==1.7.0.post1', 'awslogs',