From 20d4dd0855bbfc69408b6e2c31db5c91fdbfba32 Mon Sep 17 00:00:00 2001 From: Molly He Date: Tue, 15 Jul 2025 14:40:42 -0700 Subject: [PATCH] update config import path in template --- .../hyperpod_pytorch_job_template/v1_0/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py b/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py index 9abc986a..9415968b 100644 --- a/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py +++ b/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py @@ -1,7 +1,7 @@ from pydantic import BaseModel, ConfigDict, Field from typing import Optional, List, Dict, Union -from sagemaker.hyperpod.training.config.hyperpod_pytorch_job_config import ( - Container, +from sagemaker.hyperpod.training.config.hyperpod_pytorch_job_unified_config import ( + Containers, ReplicaSpec, Resources, RunPolicy, @@ -103,7 +103,7 @@ def to_domain(self) -> Dict: ] # Create container object - container = Container(**container_kwargs) + container = Containers(**container_kwargs) # Create pod spec kwargs spec_kwargs = {"containers": list([container])}