sagemaker-core 1.0.62__py3-none-any.whl → 2.1.1__py3-none-any.whl

sagemaker/core/compute_resource_requirements/__init__.py

@@ -0,0 +1,18 @@
+# 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 athis 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.
+"""Compute Resource Requirements needed to deploy a model"""
+from __future__ import absolute_import
+
+from sagemaker.core.compute_resource_requirements.resource_requirements import (  # noqa: F401
+    ResourceRequirements,
+)

sagemaker/core/compute_resource_requirements/resource_requirements.py

@@ -0,0 +1,94 @@
+# 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.
+"""Defines the ResourceRequirements class that configures the compute resources for the Model."""
+
+from __future__ import absolute_import
+
+import logging
+from typing import Optional, Dict
+
+from sagemaker.core.common_utils import stringify_object
+
+LOGGER = logging.getLogger("sagemaker")
+
+
+class ResourceRequirements(object):
+    """Configures the compute resources for a Model."""
+
+    def __init__(
+        self,
+        requests: Optional[Dict[str, int]] = None,
+        limits: Optional[Dict[str, int]] = None,
+    ):
+        """Initializes a ResourceRequirements object for an Amazon SageMaker inference component.
+
+        Args:
+            requests (dict): Basic resource to be requested, including num_cpus, memory (in MB),
+                accelerator_memory (in MB), copies.
+            limits (dict): Max resource limit to be requested, including memory (in MB).
+
+            Example:
+                requests = {
+                     num_cpus: 1,
+                     memory: 1024,
+                     copies: 5
+                },
+                limits = {
+                    memory: 2048
+                }
+        """
+        self.requests = requests
+        self.limits = limits
+        self.num_accelerators: Optional[int] = None
+        self.num_cpus: Optional[int] = None
+        self.min_memory: Optional[int] = None
+        self.max_memory: Optional[int] = None
+        self.copy_count = 1
+
+        if requests:
+            if "num_accelerators" in requests:
+                self.num_accelerators = requests["num_accelerators"]
+            if "num_cpus" in requests:
+                self.num_cpus = requests["num_cpus"]
+            if "memory" in requests:
+                self.min_memory = requests["memory"]
+            if "copies" in requests:
+                self.copy_count = requests["copies"]
+        if limits:
+            if "memory" in limits:
+                self.max_memory = limits["memory"]
+
+    def __str__(self) -> str:
+        """Overriding str(*) method to make more human-readable."""
+        return stringify_object(self)
+
+    def __eq__(self, other) -> bool:
+        """Implement equal method for easy unit testing."""
+        return self.requests == other.requests and self.limits == other.limits
+
+    def get_compute_resource_requirements(self) -> dict:
+        """Returns a dict of resource requirements."""
+        resource_requirements = {
+            "MinMemoryRequiredInMb": self.min_memory,
+        }
+
+        if self.max_memory:
+            resource_requirements.update({"MaxMemoryRequiredInMb": self.max_memory})
+        if self.num_cpus:
+            resource_requirements.update({"NumberOfCpuCoresRequired": self.num_cpus})
+        if self.num_accelerators:
+            resource_requirements.update(
+                {"NumberOfAcceleratorDevicesRequired": self.num_accelerators}
+            )
+
+        return resource_requirements

sagemaker/core/config/__init__.py

@@ -0,0 +1,181 @@
+# 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 athis 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 configures the default values for SageMaker Python SDK."""
+
+from __future__ import absolute_import
+from sagemaker.core.config.config import (  # noqa: F401
+    load_local_mode_config,
+    load_sagemaker_config,
+    validate_sagemaker_config,
+)
+from sagemaker.core.config.config_schema import (  # noqa: F401
+    KEY,
+    TRAINING_JOB,
+    TRAINING_JOB_INTER_CONTAINER_ENCRYPTION_PATH,
+    TRAINING_JOB_ROLE_ARN_PATH,
+    TRAINING_JOB_ENABLE_NETWORK_ISOLATION_PATH,
+    TRAINING_JOB_ENVIRONMENT_PATH,
+    TRAINING_JOB_VPC_CONFIG_PATH,
+    TRAINING_JOB_OUTPUT_DATA_CONFIG_PATH,
+    TRAINING_JOB_RESOURCE_CONFIG_PATH,
+    TRAINING_JOB_PROFILE_CONFIG_PATH,
+    TRAINING_JOB_DISABLE_PROFILER_PATH,
+    PROCESSING_JOB_INPUTS_PATH,
+    PROCESSING_JOB,
+    PROCESSING_JOB_ENVIRONMENT_PATH,
+    PROCESSING_JOB_INTER_CONTAINER_ENCRYPTION_PATH,
+    PROCESSING_JOB_ROLE_ARN_PATH,
+    PROCESSING_JOB_NETWORK_CONFIG_PATH,
+    PROCESSING_OUTPUT_CONFIG_PATH,
+    PROCESSING_JOB_PROCESSING_RESOURCES_PATH,
+    MONITORING_JOB_ENVIRONMENT_PATH,
+    MONITORING_JOB_ROLE_ARN_PATH,
+    MONITORING_JOB_VOLUME_KMS_KEY_ID_PATH,
+    MONITORING_JOB_NETWORK_CONFIG_PATH,
+    MONITORING_JOB_OUTPUT_KMS_KEY_ID_PATH,
+    MONITORING_SCHEDULE,
+    MONITORING_SCHEDULE_INTER_CONTAINER_ENCRYPTION_PATH,
+    AUTO_ML_ROLE_ARN_PATH,
+    AUTO_ML_V2_ROLE_ARN_PATH,
+    AUTO_ML_OUTPUT_CONFIG_PATH,
+    AUTO_ML_V2_OUTPUT_CONFIG_PATH,
+    AUTO_ML_JOB_CONFIG_PATH,
+    AUTO_ML_JOB,
+    AUTO_ML_JOB_V2,
+    COMPILATION_JOB_ROLE_ARN_PATH,
+    COMPILATION_JOB_OUTPUT_CONFIG_PATH,
+    COMPILATION_JOB_VPC_CONFIG_PATH,
+    COMPILATION_JOB,
+    EDGE_PACKAGING_ROLE_ARN_PATH,
+    EDGE_PACKAGING_RESOURCE_KEY_PATH,
+    EDGE_PACKAGING_OUTPUT_CONFIG_PATH,
+    EDGE_PACKAGING_JOB,
+    TRANSFORM_JOB,
+    TRANSFORM_JOB_KMS_KEY_ID_PATH,
+    TRANSFORM_OUTPUT_KMS_KEY_ID_PATH,
+    VOLUME_KMS_KEY_ID,
+    TRANSFORM_JOB_VOLUME_KMS_KEY_ID_PATH,
+    MODEL,
+    MODEL_CONTAINERS_PATH,
+    MODEL_EXECUTION_ROLE_ARN_PATH,
+    MODEL_ENABLE_NETWORK_ISOLATION_PATH,
+    MODEL_VPC_CONFIG_PATH,
+    MODEL_PACKAGE_VALIDATION_ROLE_PATH,
+    VALIDATION_ROLE,
+    VALIDATION_PROFILES,
+    MODEL_PACKAGE_INFERENCE_SPECIFICATION_CONTAINERS_PATH,
+    MODEL_PACKAGE_VALIDATION_PROFILES_PATH,
+    MODEL_PRIMARY_CONTAINER_PATH,
+    MODEL_PRIMARY_CONTAINER_ENVIRONMENT_PATH,
+    ENDPOINT_CONFIG_PRODUCTION_VARIANTS_PATH,
+    KMS_KEY_ID,
+    RESOURCE_KEY,
+    ENDPOINT_CONFIG_KMS_KEY_ID_PATH,
+    ENDPOINT_CONFIG,
+    ENDPOINT_CONFIG_DATA_CAPTURE_PATH,
+    ENDPOINT_CONFIG_ASYNC_INFERENCE_PATH,
+    ENDPOINT_CONFIG_EXECUTION_ROLE_ARN_PATH,
+    ENDPOINT_CONFIG_ENABLE_NETWORK_ISOLATION_PATH,
+    ENDPOINT_CONFIG_VPC_CONFIG_PATH,
+    ENDPOINT,
+    INFERENCE_COMPONENT,
+    ENDPOINT_TAGS_PATH,
+    SAGEMAKER,
+    FEATURE_GROUP,
+    TAGS,
+    FEATURE_GROUP_ROLE_ARN_PATH,
+    FEATURE_GROUP_ONLINE_STORE_CONFIG_PATH,
+    FEATURE_GROUP_OFFLINE_STORE_CONFIG_PATH,
+    PIPELINE_ROLE_ARN_PATH,
+    TRANSFORM_RESOURCES_VOLUME_KMS_KEY_ID_PATH,
+    EDGE_PACKAGING_KMS_KEY_ID_PATH,
+    ENDPOINT_CONFIG_ASYNC_KMS_KEY_ID_PATH,
+    PROCESSING_JOB_KMS_KEY_ID_PATH,
+    PROCESSING_JOB_SECURITY_GROUP_IDS_PATH,
+    PROCESSING_JOB_SUBNETS_PATH,
+    PROCESSING_JOB_ENABLE_NETWORK_ISOLATION_PATH,
+    PROCESSING_JOB_VOLUME_KMS_KEY_ID_PATH,
+    PIPELINE_TAGS_PATH,
+    TRAINING_JOB_VOLUME_KMS_KEY_ID_PATH,
+    TRAINING_JOB_SECURITY_GROUP_IDS_PATH,
+    TRAINING_JOB_SUBNETS_PATH,
+    TRAINING_JOB_KMS_KEY_ID_PATH,
+    FEATURE_GROUP_OFFLINE_STORE_KMS_KEY_ID_PATH,
+    FEATURE_GROUP_ONLINE_STORE_KMS_KEY_ID_PATH,
+    AUTO_ML_KMS_KEY_ID_PATH,
+    AUTO_ML_V2_KMS_KEY_ID_PATH,
+    AUTO_ML_VPC_CONFIG_PATH,
+    AUTO_ML_V2_VPC_CONFIG_PATH,
+    AUTO_ML_VOLUME_KMS_KEY_ID_PATH,
+    AUTO_ML_V2_VOLUME_KMS_KEY_ID_PATH,
+    AUTO_ML_INTER_CONTAINER_ENCRYPTION_PATH,
+    AUTO_ML_V2_INTER_CONTAINER_ENCRYPTION_PATH,
+    ENDPOINT_CONFIG_DATA_CAPTURE_KMS_KEY_ID_PATH,
+    SESSION_DEFAULT_S3_BUCKET_PATH,
+    SESSION_DEFAULT_S3_OBJECT_KEY_PREFIX_PATH,
+    MONITORING_SCHEDULE_CONFIG,
+    MONITORING_JOB_DEFINITION,
+    MONITORING_OUTPUT_CONFIG,
+    MONITORING_RESOURCES,
+    CLUSTER_CONFIG,
+    NETWORK_CONFIG,
+    ENABLE_INTER_CONTAINER_TRAFFIC_ENCRYPTION,
+    ENABLE_NETWORK_ISOLATION,
+    VPC_CONFIG,
+    SUBNETS,
+    SECURITY_GROUP_IDS,
+    ROLE_ARN,
+    VALUE,
+    OUTPUT_CONFIG,
+    DATA_CAPTURE_CONFIG,
+    PRODUCTION_VARIANTS,
+    AUTO_ML_JOB_CONFIG,
+    SECURITY_CONFIG,
+    OUTPUT_DATA_CONFIG,
+    MODEL_PACKAGE,
+    VALIDATION_SPECIFICATION,
+    TRANSFORM_JOB_DEFINITION,
+    TRANSFORM_JOB_ENVIRONMENT_PATH,
+    TRANSFORM_OUTPUT,
+    TRANSFORM_RESOURCES,
+    OFFLINE_STORE_CONFIG,
+    S3_STORAGE_CONFIG,
+    ONLINE_STORE_CONFIG,
+    PROCESSING_INPUTS,
+    DATASET_DEFINITION,
+    ATHENA_DATASET_DEFINITION,
+    REDSHIFT_DATASET_DEFINITION,
+    CLUSTER_ROLE_ARN,
+    PROCESSING_OUTPUT_CONFIG,
+    PROCESSING_RESOURCES,
+    PROFILER_CONFIG,
+    DISABLE_PROFILER,
+    RESOURCE_CONFIG,
+    EXECUTION_ROLE_ARN,
+    ASYNC_INFERENCE_CONFIG,
+    SCHEMA_VERSION,
+    PYTHON_SDK,
+    MODULES,
+    DEFAULT_S3_BUCKET,
+    DEFAULT_S3_OBJECT_KEY_PREFIX,
+    SESSION,
+    ENVIRONMENT,
+    CONTAINERS,
+    PRIMARY_CONTAINER,
+    INFERENCE_SPECIFICATION,
+    DEBUG_HOOK_CONFIG,
+    SAGEMAKER_PYTHON_SDK_LOCAL_MODE_CONFIG_SCHEMA,
+    LOCAL,
+    LOCAL_CODE,
+    CONTAINER_CONFIG,
+)

sagemaker/core/config/config.py

@@ -0,0 +1,238 @@
+# 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 configures the default values defined by the user for SageMaker Python SDK calls.
+
+It supports loading config files from the local file system and Amazon S3.
+The schema of the config file is dictated in config_schema.py in the same module.
+
+"""
+from __future__ import absolute_import, annotations
+
+import pathlib
+import os
+from typing import List
+import boto3
+import yaml
+import jsonschema
+from platformdirs import site_config_dir, user_config_dir
+from botocore.utils import merge_dicts
+from six.moves.urllib.parse import urlparse
+from sagemaker.core.config.config_schema import SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA
+from sagemaker.core.config.config_utils import (
+    non_repeating_log_factory,
+    get_sagemaker_config_logger,
+)
+
+logger = get_sagemaker_config_logger()
+log_info_function = non_repeating_log_factory(logger, "info")
+
+_APP_NAME = "sagemaker"
+# The default name of the config file.
+_CONFIG_FILE_NAME = "config.yaml"
+# The default config file location of the Administrator provided config file. This path can be
+# overridden with `SAGEMAKER_ADMIN_CONFIG_OVERRIDE` environment variable.
+_DEFAULT_ADMIN_CONFIG_FILE_PATH = os.path.join(site_config_dir(_APP_NAME), _CONFIG_FILE_NAME)
+# The default config file location of the user provided config file. This path can be
+# overridden with `SAGEMAKER_USER_CONFIG_OVERRIDE` environment variable.
+_DEFAULT_USER_CONFIG_FILE_PATH = os.path.join(user_config_dir(_APP_NAME), _CONFIG_FILE_NAME)
+# The default config file location of the local mode.
+_DEFAULT_LOCAL_MODE_CONFIG_FILE_PATH = os.path.join(
+    os.path.expanduser("~"), ".sagemaker", _CONFIG_FILE_NAME
+)
+
+ENV_VARIABLE_ADMIN_CONFIG_OVERRIDE = "SAGEMAKER_ADMIN_CONFIG_OVERRIDE"
+ENV_VARIABLE_USER_CONFIG_OVERRIDE = "SAGEMAKER_USER_CONFIG_OVERRIDE"
+
+S3_PREFIX = "s3://"
+
+
+def load_sagemaker_config(
+    additional_config_paths: List[str] = None, s3_resource=None, repeat_log=False
+) -> dict:
+    """Loads config files and merges them.
+
+    By default, this method first searches for config files in the default locations
+    defined by the SDK.
+
+    Users can override the default admin and user config file paths using the
+    ``SAGEMAKER_ADMIN_CONFIG_OVERRIDE`` and ``SAGEMAKER_USER_CONFIG_OVERRIDE`` environment
+    variables, respectively.
+
+    Additional config file paths can also be provided as a parameter.
+
+    This method then:
+        * Loads each config file, whether it is Amazon S3 or the local file system.
+        * Validates the schema of the config files.
+        * Merges the files in the same order.
+
+    This method throws exceptions in the following cases:
+        * ``jsonschema.exceptions.ValidationError``: Schema validation fails for one or more
+          config files.
+        * ``RuntimeError``: The method is unable to retrieve the list of all S3 files with the
+          same prefix or is unable to retrieve the file.
+        * ``ValueError``: There are no S3 files with the prefix when an S3 URI is provided.
+        * ``ValueError``: There is no config.yaml file in the S3 bucket when an S3 URI is
+          provided.
+        * ``ValueError``: A file doesn't exist in a path that was specified by the user as
+          part of an environment variable or additional configuration file path. This doesn't
+          include the default config file locations.
+
+    Args:
+        additional_config_paths: List of config file paths.
+            These paths can be one of the following. In the case of a directory, this method
+            searches for a ``config.yaml`` file in that directory. This method does not perform a
+            recursive search of folders in that directory.
+
+                * Local file path
+                * Local directory path
+                * S3 URI of the config file
+                * S3 URI of the directory containing the config file
+
+            Note: S3 URI follows the format ``s3://<bucket>/<Key prefix>``
+        s3_resource (boto3.resource("s3")): The Boto3 S3 resource. This is used to fetch
+            config files from S3. If it is not provided but config files are present in S3,
+            this method creates a default S3 resource. See `Boto3 Session documentation
+            <https://boto3.amazonaws.com/v1/documentation/api\
+            /latest/reference/core/session.html#boto3.session.Session.resource>`__.
+            This argument is not needed if the config files are present in the local file system.
+        repeat_log (bool): Whether the log with the same contents should be emitted.
+            Default to ``False``
+    """
+    default_config_path = os.getenv(
+        ENV_VARIABLE_ADMIN_CONFIG_OVERRIDE, _DEFAULT_ADMIN_CONFIG_FILE_PATH
+    )
+    user_config_path = os.getenv(ENV_VARIABLE_USER_CONFIG_OVERRIDE, _DEFAULT_USER_CONFIG_FILE_PATH)
+    config_paths = [default_config_path, user_config_path]
+    if additional_config_paths:
+        config_paths += additional_config_paths
+    config_paths = list(filter(lambda item: item is not None, config_paths))
+    merged_config = {}
+
+    log_info = log_info_function
+    if repeat_log:
+        log_info = logger.info
+
+    for file_path in config_paths:
+        config_from_file = {}
+        if file_path.startswith(S3_PREFIX):
+            config_from_file = _load_config_from_s3(file_path, s3_resource)
+        else:
+            try:
+                config_from_file = _load_config_from_file(file_path)
+            except ValueError as error:
+                if file_path not in (
+                    _DEFAULT_ADMIN_CONFIG_FILE_PATH,
+                    _DEFAULT_USER_CONFIG_FILE_PATH,
+                ):
+                    # Throw exception only when User provided file path is invalid.
+                    # If there are no files in the Default config file locations, don't throw
+                    # Exceptions.
+                    raise
+
+                logger.debug(error)
+        if config_from_file:
+            validate_sagemaker_config(config_from_file)
+            merge_dicts(merged_config, config_from_file)
+            log_info("Fetched defaults config from location: %s", file_path)
+        else:
+            log_info("Not applying SDK defaults from location: %s", file_path)
+
+    return merged_config
+
+
+def validate_sagemaker_config(sagemaker_config: dict = None):
+    """Validates whether a given dictionary adheres to the schema.
+
+    The schema is defined at
+    ``sagemaker.config.config_schema.SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA``.
+
+    Args:
+        sagemaker_config: A dictionary containing default values for the
+                SageMaker Python SDK. (default: None).
+    """
+    jsonschema.validate(sagemaker_config, SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA)
+
+
+def load_local_mode_config() -> dict | None:
+    """Loads the local mode config file."""
+    try:
+        content = _load_config_from_file(_DEFAULT_LOCAL_MODE_CONFIG_FILE_PATH)
+    except ValueError:
+        content = None
+
+    return content
+
+
+def _load_config_from_file(file_path: str) -> dict:
+    """Placeholder docstring"""
+    inferred_file_path = file_path
+    if os.path.isdir(file_path):
+        inferred_file_path = os.path.join(file_path, _CONFIG_FILE_NAME)
+    if not os.path.exists(inferred_file_path):
+        raise ValueError(
+            f"Unable to load the config file from the location: {file_path}"
+            f"Provide a valid file path"
+        )
+    logger.debug("Fetching defaults config from location: %s", file_path)
+    with open(inferred_file_path, "r") as f:
+        content = yaml.safe_load(f)
+    return content
+
+
+def _load_config_from_s3(s3_uri, s3_resource_for_config) -> dict:
+    """Placeholder docstring"""
+    if not s3_resource_for_config:
+        # Constructing a default Boto3 S3 Resource from a default Boto3 session.
+        boto_session = boto3.DEFAULT_SESSION or boto3.Session()
+        boto_region_name = boto_session.region_name
+        if boto_region_name is None:
+            raise ValueError(
+                "Must setup local AWS configuration with a region supported by SageMaker."
+            )
+        s3_resource_for_config = boto_session.resource("s3", region_name=boto_region_name)
+
+    logger.debug("Fetching defaults config from location: %s", s3_uri)
+    inferred_s3_uri = _get_inferred_s3_uri(s3_uri, s3_resource_for_config)
+    parsed_url = urlparse(inferred_s3_uri)
+    bucket, key_prefix = parsed_url.netloc, parsed_url.path.lstrip("/")
+    s3_object = s3_resource_for_config.Object(bucket, key_prefix)
+    s3_file_content = s3_object.get()["Body"].read()
+    return yaml.safe_load(s3_file_content.decode("utf-8"))
+
+
+def _get_inferred_s3_uri(s3_uri, s3_resource_for_config):
+    """Placeholder docstring"""
+    parsed_url = urlparse(s3_uri)
+    bucket, key_prefix = parsed_url.netloc, parsed_url.path.lstrip("/")
+    s3_bucket = s3_resource_for_config.Bucket(name=bucket)
+    s3_objects = s3_bucket.objects.filter(Prefix=key_prefix).all()
+    s3_files_with_same_prefix = [
+        "{}{}/{}".format(S3_PREFIX, bucket, s3_object.key) for s3_object in s3_objects
+    ]
+    if len(s3_files_with_same_prefix) == 0:
+        # Customer provided us with an incorrect s3 path.
+        raise ValueError("Provide a valid S3 path instead of {}".format(s3_uri))
+    if len(s3_files_with_same_prefix) > 1:
+        # Customer has provided us with a S3 URI which points to a directory
+        # search for s3://<bucket>/directory-key-prefix/config.yaml
+        inferred_s3_uri = str(pathlib.PurePosixPath(s3_uri, _CONFIG_FILE_NAME)).replace(
+            "s3:/", "s3://"
+        )
+        if inferred_s3_uri not in s3_files_with_same_prefix:
+            # We don't know which file we should be operating with.
+            raise ValueError(
+                f"Provide an S3 URI of a directory that has a {_CONFIG_FILE_NAME} file."
+            )
+        # Customer has a config.yaml present in the directory that was provided as the S3 URI
+        return inferred_s3_uri
+    return s3_uri