fix: skip p2 test in regions without p2s, freeze urllib3, and specify allow_pickle=True for numpy (#760)

There were a few test fixes here:

* skip a GPU TF script mode test in regions without p2 instances
* Freeze urllib3 version to avoid conflicts
* Specify allow_pickle=True in npy serializer object test

Author: yangaws

setup.py

@@ -29,7 +29,7 @@ def read_version():
 
 # Declare minimal set for installation
 required_packages = ['boto3>=1.9.64', 'numpy>=1.9.0', 'protobuf>=3.1', 'scipy>=0.19.0',
-                     'urllib3>=1.21', 'protobuf3-to-dict>=0.1.5', 'docker-compose>=1.23.0',
+                     'urllib3>=1.21, <1.25', 'protobuf3-to-dict>=0.1.5', 'docker-compose>=1.23.0',
                      'requests>=2.20.0, <2.21']
 
 # enum is introduced in Python 3.4. Installing enum back port

tests/integ/test_tf_script_mode.py

@@ -40,6 +40,8 @@ def instance_type(request):
     return request.param
 
 
+@pytest.mark.skipif(integ.test_region() in integ.HOSTING_NO_P2_REGIONS,
+                    reason='no ml.p2 instances in these regions')
 @pytest.mark.skipif(integ.PYTHON_VERSION != 'py3', reason="Script Mode tests are only configured to run with Python 3")
 def test_mnist(sagemaker_session, instance_type):
     estimator = TensorFlow(entry_point=SCRIPT,

tests/unit/test_predictor.py

@@ -249,7 +249,7 @@ def test_npy_serializer_object():
 
     result = npy_serializer(object)
 
-    assert np.array_equal(np.array(object), np.load(io.BytesIO(result)))
+    assert np.array_equal(np.array(object), np.load(io.BytesIO(result), allow_pickle=True))
 
 
 def test_npy_serializer_list_of_empty():