aws
diff --git a/‎.gitignore
Lines changed: 4 additions & 4 deletions b/‎.gitignore
Lines changed: 4 additions & 4 deletions
diff --git a/‎CHANGELOG.md
Lines changed: 33 additions & 0 deletions b/‎CHANGELOG.md
Lines changed: 33 additions & 0 deletions
diff --git a/‎VERSION
Lines changed: 1 addition & 1 deletion b/‎VERSION
Lines changed: 1 addition & 1 deletion
diff --git a/‎doc/overview.rst
Lines changed: 5 additions & 0 deletions b/‎doc/overview.rst
Lines changed: 5 additions & 0 deletions
diff --git a/‎doc/requirements.txt
Lines changed: 1 addition & 1 deletion b/‎doc/requirements.txt
Lines changed: 1 addition & 1 deletion
diff --git a/‎src/sagemaker/batch_inference/__init__.py renamed to ‎legacy/__init__.py b/‎src/sagemaker/batch_inference/__init__.py renamed to ‎legacy/__init__.py
diff --git a/‎src/sagemaker/modules/train/sm_recipes/__init__.py renamed to ‎legacy/src/__init__.py b/‎src/sagemaker/modules/train/sm_recipes/__init__.py renamed to ‎legacy/src/__init__.py
diff --git a/‎src/sagemaker/__init__.py renamed to ‎legacy/src/sagemaker/__init__.py b/‎src/sagemaker/__init__.py renamed to ‎legacy/src/sagemaker/__init__.py
diff --git a/‎legacy/src/sagemaker/_studio.py
Lines changed: 116 additions & 0 deletions b/‎legacy/src/sagemaker/_studio.py
Lines changed: 116 additions & 0 deletions
diff --git a/‎legacy/src/sagemaker/accept_types.py
Lines changed: 131 additions & 0 deletions b/‎legacy/src/sagemaker/accept_types.py
Lines changed: 131 additions & 0 deletions
@@ -32,9 +32,9 @@ env/
 .python-version
 *.html
 **/_repack_script_launcher.sh
-src/sagemaker/modules/train/container_drivers/sm_train.sh
-src/sagemaker/modules/train/container_drivers/sourcecode.json
-src/sagemaker/modules/train/container_drivers/distributed.json
+legacy/src/sagemaker/modules/train/container_drivers/sm_train.sh
+legacy/src/sagemaker/modules/train/container_drivers/sourcecode.json
+legacy/src/sagemaker/modules/train/container_drivers/distributed.json
 tests/data/**/_repack_model.py
 tests/data/experiment/sagemaker-dev-1.0.tar.gz
-src/sagemaker/serve/tmp_workspace
+legacy/src/sagemaker/serve/tmp_workspace
@@ -1,5 +1,38 @@
 # Changelog
 
+## v2.242.0 (2025-03-14)
+
+### Features
+
+ * add integ tests for training JumpStart models in private hub
+
+### Bug Fixes and Other Changes
+
+ * Torch upgrade
+ * Prevent RunContext overlap between test_run tests
+ * remove s3 output location requirement from hub class init
+ * Fixing Pytorch training python version in tests
+ * update image_uri_configs  03-11-2025 07:18:09 PST
+ * resolve infinite loop in _find_config on Windows systems
+ * pipeline definition function doc update
+
+## v2.241.0 (2025-03-06)
+
+### Features
+
+ * Make DistributedConfig Extensible
+ * support training for JumpStart model references as part of Curated Hub Phase 2
+ * Allow ModelTrainer to accept hyperparameters file
+
+### Bug Fixes and Other Changes
+
+ * Skip tests with deprecated instance type
+ * Ensure Model.is_repack() returns a boolean
+ * Fix error when there is no session to call _create_model_request()
+ * Use sagemaker session's s3_resource in download_folder
+ * Added check for the presence of model package group before creating one
+ * Fix key error in _send_metrics()
+
 ## v2.240.0 (2025-02-25)
 
 ### Features
 
@@ -1 +1 @@
-2.240.1.dev0
+2.242.1.dev0
@@ -30,6 +30,11 @@ To train a model by using the SageMaker Python SDK, you:
 
 After you train a model, you can save it, and then serve the model as an endpoint to get real-time inferences or get inferences for an entire dataset by using batch transform.
 
+
+Important Note:
+
+*  When using torch to load Models, it is recommended to use version torch>=2.6.0 and torchvision>=0.17.0
+
 Prepare a Training script
 =========================
 
 
@@ -2,7 +2,7 @@ sphinx==5.1.1
 sphinx-rtd-theme==0.5.0
 docutils==0.15.2
 packaging==20.9
-jinja2==3.1.4
+jinja2==3.1.6
 schema==0.7.5
 accelerate>=0.24.1,<=0.27.0
 graphene<4.0
@@ -0,0 +1,116 @@
+# Copyright 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://aws.amazon.com/apache2.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.
+"""Provides internal tooling for studio environments."""
+from __future__ import absolute_import
+
+import json
+import logging
+
+from pathlib import Path
+
+STUDIO_PROJECT_CONFIG = ".sagemaker-code-config"
+
+logger = logging.getLogger(__name__)
+
+
+def _append_project_tags(tags=None, working_dir=None):
+    """Appends the project tag to the list of tags, if it exists.
+
+    Args:
+        working_dir: the working directory to start looking.
+        tags: the list of tags to append to.
+
+    Returns:
+        A possibly extended list of tags that includes the project id.
+    """
+    path = _find_config(working_dir)
+    if path is None:
+        return tags
+
+    config = _load_config(path)
+    if config is None:
+        return tags
+
+    additional_tags = _parse_tags(config)
+    if additional_tags is None:
+        return tags
+
+    all_tags = tags or []
+    additional_tags = [tag for tag in additional_tags if tag not in all_tags]
+    all_tags.extend(additional_tags)
+
+    return all_tags
+
+
+def _find_config(working_dir=None):
+    """Gets project config on SageMaker Studio platforms, if it exists.
+
+    Args:
+        working_dir: the working directory to start looking.
+
+    Returns:
+        The project config path, if it exists. Otherwise None.
+    """
+    try:
+        wd = Path(working_dir) if working_dir else Path.cwd()
+
+        path = None
+
+        # Get the root of the current working directory for both Windows and Unix-like systems
+        root = Path(wd.anchor)
+        while path is None and wd != root:
+            candidate = wd / STUDIO_PROJECT_CONFIG
+            if Path.exists(candidate):
+                path = candidate
+            wd = wd.parent
+
+        return path
+    except Exception as e:  # pylint: disable=W0703
+        logger.debug("Could not find the studio project config. %s", e)
+
+
+def _load_config(path):
+    """Parse out the projectId attribute if it exists at path.
+
+    Args:
+        path: path to project config
+
+    Returns:
+        Project config Json, or None if it does not exist.
+    """
+    try:
+        with open(path, "r") as f:
+            content = f.read().strip()
+        config = json.loads(content)
+
+        return config
+    except Exception as e:  # pylint: disable=W0703
+        logger.debug("Could not load project config. %s", e)
+
+
+def _parse_tags(config):
+    """Parse out appropriate attributes and formats as tags.
+
+    Args:
+        config: project config dict
+
+    Returns:
+        List of tags
+    """
+    try:
+        return [
+            {"Key": "sagemaker:project-id", "Value": config["sagemakerProjectId"]},
+            {"Key": "sagemaker:project-name", "Value": config["sagemakerProjectName"]},
+        ]
+    except Exception as e:  # pylint: disable=W0703
+        logger.debug("Could not parse project config. %s", e)
@@ -0,0 +1,131 @@
+# Copyright 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://aws.amazon.com/apache2.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.
+"""This module is for SageMaker accept types."""
+from __future__ import absolute_import
+from typing import List, Optional
+
+from sagemaker.jumpstart import artifacts, utils as jumpstart_utils
+from sagemaker.jumpstart.constants import DEFAULT_JUMPSTART_SAGEMAKER_SESSION
+from sagemaker.jumpstart.enums import JumpStartModelType
+from sagemaker.session import Session
+
+
+def retrieve_options(
+    region: Optional[str] = None,
+    model_id: Optional[str] = None,
+    model_version: Optional[str] = None,
+    hub_arn: Optional[str] = None,
+    tolerate_vulnerable_model: bool = False,
+    tolerate_deprecated_model: bool = False,
+    sagemaker_session: Session = DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
+) -> List[str]:
+    """Retrieves the supported accept types for the model matching the given arguments.
+
+    Args:
+        region (str): The AWS Region for which to retrieve the supported accept types.
+            Defaults to ``None``.
+        model_id (str): The model ID of the model for which to
+            retrieve the supported accept types. (Default: None).
+        model_version (str): The version of the model for which to retrieve the
+            supported accept types. (Default: None).
+        hub_arn (str): The arn of the SageMaker Hub for which to retrieve
+            model details from. (Default: None).
+        tolerate_vulnerable_model (bool): True if vulnerable versions of model
+            specifications should be tolerated (exception not raised). If False, raises an
+            exception if the script used by this version of the model has dependencies with known
+            security vulnerabilities. (Default: False).
+        tolerate_deprecated_model (bool): True if deprecated models should be tolerated
+            (exception not raised). False if these models should raise an exception.
+            (Default: False).
+        sagemaker_session (sagemaker.session.Session): A SageMaker Session
+            object, used for SageMaker interactions. If not
+            specified, one is created using the default AWS configuration
+            chain. (Default: sagemaker.jumpstart.constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION).
+    Returns:
+        list: The supported accept types to use for the model.
+
+    Raises:
+        ValueError: If the combination of arguments specified is not supported.
+    """
+    if not jumpstart_utils.is_jumpstart_model_input(model_id, model_version):
+        raise ValueError(
+            "Must specify JumpStart `model_id` and `model_version` when retrieving accept types."
+        )
+
+    return artifacts._retrieve_supported_accept_types(
+        model_id=model_id,
+        model_version=model_version,
+        hub_arn=hub_arn,
+        region=region,
+        tolerate_vulnerable_model=tolerate_vulnerable_model,
+        tolerate_deprecated_model=tolerate_deprecated_model,
+        sagemaker_session=sagemaker_session,
+    )
+
+
+def retrieve_default(
+    region: Optional[str] = None,
+    model_id: Optional[str] = None,
+    model_version: Optional[str] = None,
+    hub_arn: Optional[str] = None,
+    tolerate_vulnerable_model: bool = False,
+    tolerate_deprecated_model: bool = False,
+    sagemaker_session: Session = DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
+    model_type: JumpStartModelType = JumpStartModelType.OPEN_WEIGHTS,
+    config_name: Optional[str] = None,
+) -> str:
+    """Retrieves the default accept type for the model matching the given arguments.
+
+    Args:
+        region (str): The AWS Region for which to retrieve the default accept type.
+            Defaults to ``None``.
+        model_id (str): The model ID of the model for which to
+            retrieve the default accept type. (Default: None).
+        model_version (str): The version of the model for which to retrieve the
+            default accept type. (Default: None).
+        hub_arn (str): The arn of the SageMaker Hub for which to retrieve
+            model details from. (Default: None).
+        tolerate_vulnerable_model (bool): True if vulnerable versions of model
+            specifications should be tolerated (exception not raised). If False, raises an
+            exception if the script used by this version of the model has dependencies with known
+            security vulnerabilities. (Default: False).
+        tolerate_deprecated_model (bool): True if deprecated models should be tolerated
+            (exception not raised). False if these models should raise an exception.
+            (Default: False).
+        sagemaker_session (sagemaker.session.Session): A SageMaker Session
+            object, used for SageMaker interactions. If not
+            specified, one is created using the default AWS configuration
+            chain. (Default: sagemaker.jumpstart.constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION).
+        config_name (Optional[str]): Name of the JumpStart Model config to apply. (Default: None).
+    Returns:
+        str: The default accept type to use for the model.
+
+    Raises:
+        ValueError: If the combination of arguments specified is not supported.
+    """
+    if not jumpstart_utils.is_jumpstart_model_input(model_id, model_version):
+        raise ValueError(
+            "Must specify JumpStart `model_id` and `model_version` when retrieving accept types."
+        )
+
+    return artifacts._retrieve_default_accept_type(
+        model_id=model_id,
+        model_version=model_version,
+        hub_arn=hub_arn,
+        region=region,
+        tolerate_vulnerable_model=tolerate_vulnerable_model,
+        tolerate_deprecated_model=tolerate_deprecated_model,
+        sagemaker_session=sagemaker_session,
+        model_type=model_type,
+        config_name=config_name,
+    )