upgrade to latest sagemaker-experiments

Author: danabens

docker/1.15.0/py3/Dockerfile.cpu

@@ -100,7 +100,7 @@ RUN pip install --no-cache-dir -U \
     keras==2.3.1 \
     requests==2.22.0 \
     smdebug==0.5.0.post0 \
-    sagemaker-experiments==0.1.3 \
+    sagemaker-experiments==0.1.7 \
     mpi4py==3.0.2 \
     "cryptography>=2.3" \
     "sagemaker-tensorflow>=1.15,<1.16" \

docker/1.15.0/py3/Dockerfile.gpu

@@ -137,7 +137,7 @@ RUN pip install --no-cache-dir -U \
     requests==2.22.0 \
     keras==2.3.1 \
     smdebug==0.5.0.post0 \
-    sagemaker-experiments==0.1.3 \
+    sagemaker-experiments==0.1.7 \
     mpi4py==3.0.2 \
     "cryptography>=2.3" \
     "sagemaker-tensorflow>=1.15,<1.16" \

setup.py

@@ -59,7 +59,7 @@ def read_version():
         'test': ['tox', 'flake8', 'pytest', 'pytest-cov', 'pytest-xdist', 'mock',
                  'sagemaker==1.50.1', 'tensorflow<2.0', 'docker-compose', 'boto3==1.10.50',
                  'six==1.13.0', 'python-dateutil>=2.1,<2.8.1', 'botocore==1.13.50',
-                 'requests-mock', 'awscli==1.16.314'],
+                 'requests-mock', 'awscli==1.16.314', 'sagemaker-experiments==0.1.7'],
         'benchmark': ['click']
     },
 )

test/integration/conftest.py

@@ -116,3 +116,10 @@ def docker_image(docker_base_name, tag):
 def ecr_image(account_id, docker_base_name, tag, region):
     return '{}.dkr.ecr.{}.amazonaws.com/{}:{}'.format(
         account_id, region, docker_base_name, tag)
+
+
+@pytest.fixture(autouse=True)
+def skip_py2_containers(request, tag):
+    if request.node.get_closest_marker('skip_py2_containers'):
+        if 'py2' in tag:
+            pytest.skip('Skipping python2 container with tag {}'.format(tag))

test/integration/sagemaker/test_experiments.py

@@ -0,0 +1,37 @@
+# Copyright 2018-2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License").
+# You may not use this file except in compliance with the License.
+# A copy of the License is located at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the "license" file accompanying this file. This file is distributed
+# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+# express or implied. See the License for the specific language governing
+# permissions and limitations under the License.
+from __future__ import absolute_import
+
+import time
+
+import pytest
+from smexperiments.experiment import Experiment
+
+
+@pytest.mark.skip_py2_containers
+@pytest.mark.skip_test_in_region
+def test_experiments(sagemaker_session, ecr_image, instance_type, instance_count):
+    sm = sagemaker_session.sagemaker_client
+
+    experiment_name = f"tensorflow-container-integ-test-{int(time.time())}"
+
+    created_experiment = Experiment.create(
+        experiment_name=experiment_name,
+        description="Integration test experiment from DLC",
+        sagemaker_boto_client=sm
+    )
+    print(created_experiment)
+
+    experiment = Experiment.load(experiment_name=experiment_name, sagemaker_boto_client=sm)
+
+    experiment.delete()