Merge pull request #4 from jaipreet-s/master

Add compatibility with SageMaker SDK v2
aws-samples · Oct 30, 2020 · e6ce9cd · e6ce9cd
2 parents 464e83f + 82e3d7d
commit e6ce9cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -184,7 +184,7 @@ From a "System Terminal" in SageMaker Studio
 ```bash
 export DEV_S3_PATH_PREFIX=s3://path/to/location
 aws s3 sync ${DEV_S3_PATH_PREFIX}/sagemaker-docker-build/dist . 
-pip install sagemaker-docker-build-x.y.tar.gz
+pip install sagemaker_studio_image_build-x.y.z.tar.gz
 ```
 
 ## Security

diff --git a/setup.py b/setup.py
@@ -3,11 +3,11 @@
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-required_packages = ["boto3>=1.10.44, < 2.0", "sagemaker < 2.0"]
+required_packages = ["boto3>=1.10.44, < 2.0", "sagemaker < 3.0"]
 
 setuptools.setup(
     name="sagemaker_studio_image_build",
-    version="0.3.1",
+    version="0.4.0",
     author="Amazon Web Services",
     description="Build Docker Images in Amazon SageMaker Studio using AWS CodeBuild",
     long_description=long_description,
@@ -20,7 +20,6 @@
         "Intended Audience :: Science/Research",
         "Programming Language :: Python",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.5",
         "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
     ],