Merge branch 'master' into add_dlc_training_new_regions

Author: saimidu

CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## v2.135.0 (2023-02-23)
+
+### Features
+
+ * Add DLC accounts for MEL Region
+ * allow use of short lived creds for local container
+
+### Bug Fixes and Other Changes
+
+ * update lambda function when function arn is provided
+
 ## v2.134.1 (2023-02-22)
 
 ### Bug Fixes and Other Changes

README.rst

@@ -126,7 +126,7 @@ To run the unit tests with tox, run:
 
     tox tests/unit
 
-**Integrations tests**
+**Integration tests**
 
 To run the integration tests, the following prerequisites must be met
 

VERSION

@@ -1 +1 @@
-2.134.2.dev0
+2.135.1.dev0

requirements/extras/test_requirements.txt

@@ -13,7 +13,7 @@ black==22.3.0
 stopit==1.1.2
 # Update tox.ini to have correct version of airflow constraints file
 apache-airflow==2.4.1
-apache-airflow-providers-amazon==4.0.0
+apache-airflow-providers-amazon==7.2.1
 attrs==22.1.0
 fabric==2.6.0
 requests==2.27.1

src/sagemaker/feature_store/feature_group.py

@@ -473,11 +473,12 @@ class FeatureGroup:
     Attributes:
         name (str): name of the FeatureGroup instance.
         sagemaker_session (Session): session instance to perform boto calls.
+            If None, a new Session will be created.
         feature_definitions (Sequence[FeatureDefinition]): list of FeatureDefinitions.
     """
 
     name: str = attr.ib(factory=str)
-    sagemaker_session: Session = attr.ib(default=Session)
+    sagemaker_session: Session = attr.ib(factory=Session)
     feature_definitions: Sequence[FeatureDefinition] = attr.ib(factory=list)
 
     _INTEGER_TYPES = [

src/sagemaker/git_utils.py

@@ -174,7 +174,7 @@ def _clone_command_for_github_like(git_config, dest_dir):
         CalledProcessError: If failed to clone git repo.
     """
     is_https = git_config["repo"].startswith("https://")
-    is_ssh = git_config["repo"].startswith("git@")
+    is_ssh = git_config["repo"].startswith("git@") or git_config["repo"].startswith("ssh://")
     if not is_https and not is_ssh:
         raise ValueError("Invalid Git url provided.")
     if is_ssh:
@@ -277,12 +277,16 @@ def _run_clone_command(repo_url, dest_dir):
     if repo_url.startswith("https://"):
         my_env["GIT_TERMINAL_PROMPT"] = "0"
         subprocess.check_call(["git", "clone", repo_url, dest_dir], env=my_env)
-    elif repo_url.startswith("git@"):
-        with tempfile.NamedTemporaryFile() as sshnoprompt:
-            with open(sshnoprompt.name, "w") as write_pipe:
-                write_pipe.write("ssh -oBatchMode=yes $@")
-            os.chmod(sshnoprompt.name, 0o511)
-            my_env["GIT_SSH"] = sshnoprompt.name
+    elif repo_url.startswith("git@") or repo_url.startswith("ssh://"):
+        try:
+            with tempfile.NamedTemporaryFile() as sshnoprompt:
+                with open(sshnoprompt.name, "w") as write_pipe:
+                    write_pipe.write("ssh -oBatchMode=yes $@")
+                os.chmod(sshnoprompt.name, 0o511)
+                my_env["GIT_SSH"] = sshnoprompt.name
+                subprocess.check_call(["git", "clone", repo_url, dest_dir], env=my_env)
+        except subprocess.CalledProcessError:
+            del my_env["GIT_SSH"]
             subprocess.check_call(["git", "clone", repo_url, dest_dir], env=my_env)
 
 

tests/unit/sagemaker/image_uris/test_graviton.py

@@ -27,7 +27,7 @@
     "r6g",
     "r6gd",
 ]
-GRAVITON_ALGOS = ("tensorflow", "pytotch")
+GRAVITON_ALGOS = ("tensorflow", "pytorch")
 GRAVITON_INSTANCE_TYPES = [
     "ml.m6g.4xlarge",
     "ml.m6gd.2xlarge",