Skip to content

Commit d2bd3c2

Browse files
Change default container name in pytorch template (#220)
* add metadata_name argument to js and custom endpoint to match with SDK * fix integ * change container name in pytorch template * update v1_0 too * update default container name for pytorch job template
1 parent 853dfa8 commit d2bd3c2

File tree

2 files changed

+2
-2
lines changed
  • hyperpod-pytorch-job-template/hyperpod_pytorch_job_template

2 files changed

+2
-2
lines changed

hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ def to_domain(self) -> Dict:
233233

234234
# Create container with required fields
235235
container_kwargs = {
236-
"name": "container-name",
236+
"name": "pytorch-job-container",
237237
"image": self.image,
238238
"resources": Resources(
239239
requests={"nvidia.com/gpu": "0"},

hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_1/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ def to_domain(self) -> Dict:
303303

304304
# Create container with required fields
305305
container_kwargs = {
306-
"name": "container-name",
306+
"name": "pytorch-job-container",
307307
"image": self.image,
308308
"resources": Resources(
309309
requests=requests_value,

0 commit comments

Comments
 (0)