Release 2026-03-19

Release 2026-03-19
* Fixed incorrect input data upload format for Nova 1.0 SageMaker Training Jobs

Author: balajiaru

src/amzn_nova_forge/__version__.py

@@ -11,4 +11,4 @@
 # 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.
-VERSION = "1.3.16"  # pragma: no cover
+VERSION = "1.3.17"  # pragma: no cover

src/amzn_nova_forge/manager/runtime_manager.py

@@ -27,7 +27,11 @@
 from botocore.exceptions import ClientError
 from sagemaker.ai_registry.dataset import DataSet
 from sagemaker.core.helper.session_helper import Session, get_execution_role
-from sagemaker.core.shapes import OutputDataConfig, TensorBoardOutputConfig
+from sagemaker.core.shapes import (
+    OutputDataConfig,
+    S3DataSource,
+    TensorBoardOutputConfig,
+)
 from sagemaker.core.training.configs import (
     Compute,
     InputData,
@@ -290,7 +294,12 @@ def execute(self, job_config: JobConfig) -> str:
             # https://docs.aws.amazon.com/sagemaker/latest/dg/nova-model-evaluation.html#nova-model-evaluation-notebook
             if job_config.data_s3_path:
                 input_data = InputData(
-                    channel_name="train", data_source=job_config.data_s3_path
+                    channel_name="train",
+                    data_source=S3DataSource(
+                        s3_uri=job_config.data_s3_path,
+                        s3_data_type=job_config.input_s3_data_type,
+                        s3_data_distribution_type="FullyReplicated",
+                    ),
                 )
                 trainer_config["input_data_config"] = [input_data]
 

src/amzn_nova_forge/util/recipe.py

@@ -793,14 +793,14 @@ def load_recipe_templates(
             _, _, image_uri = download_templates_from_local(
                 recipe_metadata=recipe_metadata
             )
-        
+
         # Override image URI if provided
         if image_uri_override:
             from amzn_nova_forge.validation.validator import Validator
 
             Validator.validate_ecr_image_uri(image_uri_override)
             image_uri = image_uri_override
-        
+
         return recipe_metadata, recipe_template, overrides_template, image_uri
 
     # For non-Bedrock platforms, instance_type is required