fix: fix integ test errors when running with py2 (#213)

* this change fixes module import errors in the test directory when running with Python2.7
* reduce max training steps in the mnist test from 1000 to 200 in order to shorten test runtime

Author: icywang86rui

test/integration/local/test_horovod.py

@@ -19,7 +19,7 @@
 import pytest
 from sagemaker.tensorflow import TensorFlow
 
-from test.integration.sagemaker.utils import processor, py_version  # noqa: F401
+from test.integration.utils import processor, py_version  # noqa: F401
 
 RESOURCE_PATH = os.path.join(os.path.dirname(__file__), '..', '..', 'resources')
 

test/integration/local/test_keras.py

@@ -20,7 +20,7 @@
 from sagemaker.tensorflow import serving, TensorFlow
 
 from test.integration import RESOURCE_PATH
-from test.integration.sagemaker.utils import processor, py_version  # noqa: F401
+from test.integration.utils import processor, py_version  # noqa: F401
 
 
 logging.basicConfig(level=logging.DEBUG)

test/integration/local/test_training.py

@@ -18,7 +18,7 @@
 import pytest
 from sagemaker.tensorflow import TensorFlow
 
-from test.integration.sagemaker.utils import processor, py_version  # noqa: F401
+from test.integration.utils import processor, py_version  # noqa: F401
 
 RESOURCE_PATH = os.path.join(os.path.dirname(__file__), '..', '..', 'resources')
 TF_CHECKPOINT_FILES = ['graph.pbtxt', 'model.ckpt-0.index', 'model.ckpt-0.meta']

test/integration/sagemaker/test_horovod.py

@@ -17,7 +17,7 @@
 import sagemaker
 from sagemaker.tensorflow import TensorFlow
 
-from utils import processor, py_version, unique_name_from_base  # noqa: F401
+from test.integration.utils import processor, py_version, unique_name_from_base  # noqa: F401
 
 RESOURCE_PATH = os.path.join(os.path.dirname(__file__), '..', '..', 'resources')
 

test/integration/sagemaker/test_mnist.py

@@ -19,7 +19,7 @@
 from sagemaker.tensorflow import TensorFlow
 from six.moves.urllib.parse import urlparse
 
-from utils import processor, py_version, unique_name_from_base  # noqa: F401
+from test.integration.utils import processor, py_version, unique_name_from_base  # noqa: F401
 
 
 @pytest.mark.deploy_test

test/integration/sagemaker/test_tuning_model_dir.py

@@ -17,7 +17,7 @@
 from sagemaker.tensorflow import TensorFlow
 from sagemaker.tuner import HyperparameterTuner, IntegerParameter
 
-from utils import processor, py_version, unique_name_from_base  # noqa: F401
+from test.integration.utils import processor, py_version, unique_name_from_base  # noqa: F401
 
 
 def test_model_dir_with_training_job_name(sagemaker_session, ecr_image, instance_type, framework_version):

test/integration/sagemaker/utils.py renamed to test/integration/utils.py

@@ -115,8 +115,8 @@ def _parse_args():
     parser = argparse.ArgumentParser()
     parser.add_argument('--train', type=str, default=os.environ['SM_CHANNEL_TRAINING'])
     parser.add_argument('--model_dir', type=str)
-    parser.add_argument('--max-steps', type=int, default=1000)
-    parser.add_argument('--save-checkpoint-steps', type=int, default=1000)
+    parser.add_argument('--max-steps', type=int, default=200)
+    parser.add_argument('--save-checkpoint-steps', type=int, default=200)
     parser.add_argument('--throttle-secs', type=int, default=60)
     parser.add_argument('--hosts', type=list, default=json.loads(os.environ['SM_HOSTS']))
     parser.add_argument('--current-host', type=str, default=os.environ['SM_CURRENT_HOST'])