From e04009ed1dca26a4141a27a4addbf25e9500a0d4 Mon Sep 17 00:00:00 2001 From: Molly He Date: Tue, 3 Feb 2026 15:34:35 -0800 Subject: [PATCH] Fix: Update s3 data type for fine-tuning jobs to support s3 uri --- .../src/sagemaker/train/common_utils/finetune_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py b/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py index 3fd17c3ac0..2a3c069b98 100644 --- a/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py +++ b/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py @@ -400,7 +400,7 @@ def _create_input_channels(dataset: str, content_type: Optional[str] = None, data_source = DataSource( s3_data_source={ "s3_uri": dataset, - "s3_data_type": "S3Prefix", + "s3_data_type": "Converse", "s3_data_distribution_type": "FullyReplicated" } ) @@ -639,7 +639,7 @@ def _convert_input_data_to_channels(input_data_config ): data_source = DataSource( s3_data_source={ "s3_uri": input_data.data_source, - "s3_data_type": "S3Prefix", + "s3_data_type": "Converse", "s3_data_distribution_type": "FullyReplicated" } )