diff --git a/buildspec-release.yml b/buildspec-release.yml
index 698c7945..46413129 100644
--- a/buildspec-release.yml
+++ b/buildspec-release.yml
@@ -12,7 +12,7 @@ phases:
       # run unit tests
       - AWS_ACCESS_KEY_ID= AWS_SECRET_ACCESS_KEY= AWS_SESSION_TOKEN=
         AWS_CONTAINER_CREDENTIALS_RELATIVE_URI= AWS_DEFAULT_REGION=
-        tox -e py36 -- test/unit
+        tox -e py36,py37 -- test/unit
 
       # run local integ tests
       #- $(aws ecr get-login --no-include-email --region us-west-2)
diff --git a/buildspec-toolkit.yml b/buildspec-toolkit.yml
index 0bcc7a12..cf4c58b6 100644
--- a/buildspec-toolkit.yml
+++ b/buildspec-toolkit.yml
@@ -33,7 +33,7 @@ phases:
       - tox -e flake8,twine
 
       # run unit tests
-      - tox -e py36 test-toolkit/unit
+      - tox -e py36,py37 test-toolkit/unit
 
       # define tags
       - GENERIC_TAG="$FRAMEWORK_VERSION-pytorch-$BUILD_ID"
diff --git a/setup.py b/setup.py
index 548e0720..54c0bf31 100644
--- a/setup.py
+++ b/setup.py
@@ -45,11 +45,12 @@ def read(fname):
         "Programming Language :: Python",
         'Programming Language :: Python :: 2.7',
         'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
     ],
 
     # We don't declare our dependency on torch here because we build with
     # different packages for different variants
-    install_requires=['numpy', 'retrying', 'sagemaker-inference>=1.2.2'],
+    install_requires=['numpy', 'retrying', 'sagemaker-inference>=1.3.1'],
     extras_require={
         'test': ['boto3==1.10.32', 'coverage==4.5.3', 'docker-compose==1.23.2', 'flake8==3.7.7', 'Flask==1.1.1',
                  'mock==2.0.0', 'pytest==4.4.0', 'pytest-cov==2.7.1', 'pytest-xdist==1.28.0', 'PyYAML==3.10',
diff --git a/tox.ini b/tox.ini
index 937ff546..784db343 100644
--- a/tox.ini
+++ b/tox.ini
@@ -4,7 +4,7 @@
 # and then run "tox" from this directory.
 
 [tox]
-envlist = py36,flake8
+envlist = flake8,py36,py37
 skip_missing_interpreters = False
 
 [flake8]