fix: resolve pluggy version conflict (#211)

Author: icywang86rui

setup.py

@@ -56,7 +56,7 @@ def read_version():
     install_requires=['sagemaker-containers>=2.4.6', 'numpy', 'scipy', 'sklearn',
                       'pandas', 'Pillow', 'h5py'],
     extras_require={
-        'test': ['tox', 'flake8', 'pytest', 'pytest-cov', 'pytest-xdist', 'mock',
+        'test': ['tox', 'flake8', 'pytest==4.4.1', 'pytest-cov', 'pytest-xdist', 'mock',
                  'sagemaker==1.19.1', 'tensorflow', 'docker-compose', 'botocore>=1.12.140'],
         'benchmark': ['click']
     },

test/integration/sagemaker/test_mnist.py

@@ -18,7 +18,6 @@
 from sagemaker.tensorflow import TensorFlow
 from six.moves.urllib.parse import urlparse
 
-from sagemaker_tensorflow_container.training import SAGEMAKER_PARAMETER_SERVER_ENABLED
 from utils import processor, py_version, unique_name_from_base  # noqa: F401
 
 
@@ -63,7 +62,7 @@ def test_distributed_mnist_ps(sagemaker_session, ecr_image, instance_type, frame
     script = os.path.join(resource_path, 'mnist', 'mnist_estimator.py')
     estimator = TensorFlow(entry_point=script,
                            role='SageMakerRole',
-                           hyperparameters={SAGEMAKER_PARAMETER_SERVER_ENABLED: True},
+                           hyperparameters={'sagemaker_parameter_server_enabled': True},
                            train_instance_count=2,
                            train_instance_type=instance_type,
                            sagemaker_session=sagemaker_session,