change: Add py37 to sm tests (#361)

Author: saimidu

test/integration/conftest.py

@@ -38,7 +38,7 @@ def pytest_addoption(parser):
     parser.addoption('--region', default='us-west-2')
     parser.addoption('--framework-version', default=TensorFlow.LATEST_VERSION)
     parser.addoption('--processor', default='cpu', choices=['cpu', 'gpu', 'cpu,gpu'])
-    parser.addoption('--py-version', default='3', choices=['2', '3', '2,3'])
+    parser.addoption('--py-version', default='3', choices=['2', '3', '2,3', '37'])
     parser.addoption('--account-id', default='142577830533')
     parser.addoption('--instance-type', default=None)
 

test/integration/local/test_training.py

@@ -28,6 +28,8 @@
 def py_full_version(py_version):  # noqa: F811
     if py_version == '2':
         return '2.7'
+    elif py_version == '37':
+        return '3.7'
     else:
         return '3.6'