Skip to content

Commit 670bf6a

Browse files
authored
Merge branch 'master' into fix-jumpstart-test
2 parents a5b2d62 + a0c39ef commit 670bf6a

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

src/sagemaker/image_uri_config/spark.json

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
"ap-southeast-2": "440695851116",
2121
"ap-southeast-3": "800295151634",
2222
"ap-southeast-4": "819679513684",
23+
"ap-southeast-5": "841784149062",
2324
"ca-central-1": "446299261295",
2425
"ca-west-1": "000907499111",
2526
"cn-north-1": "671472414489",
@@ -61,6 +62,7 @@
6162
"ap-southeast-2": "440695851116",
6263
"ap-southeast-3": "800295151634",
6364
"ap-southeast-4": "819679513684",
65+
"ap-southeast-5": "841784149062",
6466
"ca-central-1": "446299261295",
6567
"ca-west-1": "000907499111",
6668
"cn-north-1": "671472414489",
@@ -102,6 +104,7 @@
102104
"ap-southeast-2": "440695851116",
103105
"ap-southeast-3": "800295151634",
104106
"ap-southeast-4": "819679513684",
107+
"ap-southeast-5": "841784149062",
105108
"ca-central-1": "446299261295",
106109
"ca-west-1": "000907499111",
107110
"cn-north-1": "671472414489",
@@ -143,6 +146,7 @@
143146
"ap-southeast-2": "440695851116",
144147
"ap-southeast-3": "800295151634",
145148
"ap-southeast-4": "819679513684",
149+
"ap-southeast-5": "841784149062",
146150
"ca-central-1": "446299261295",
147151
"ca-west-1": "000907499111",
148152
"cn-north-1": "671472414489",
@@ -184,6 +188,7 @@
184188
"ap-southeast-2": "440695851116",
185189
"ap-southeast-3": "800295151634",
186190
"ap-southeast-4": "819679513684",
191+
"ap-southeast-5": "841784149062",
187192
"ca-central-1": "446299261295",
188193
"ca-west-1": "000907499111",
189194
"cn-north-1": "671472414489",

src/sagemaker/local/image.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,12 @@ def write_processing_config_files(
473473
"""
474474
config_path = os.path.join(self.container_root, host, "config")
475475

476-
resource_config = {"current_host": host, "hosts": self.hosts}
476+
resource_config = {
477+
"current_host": host,
478+
"hosts": self.hosts,
479+
"network_interface_name": "eth0",
480+
"current_instance_type": self.instance_type,
481+
}
477482
_write_json_file(os.path.join(config_path, "resourceconfig.json"), resource_config)
478483

479484
processing_job_config = {
@@ -519,7 +524,12 @@ def write_config_files(self, host, hyperparameters, input_data_config):
519524
"""
520525
config_path = os.path.join(self.container_root, host, "input", "config")
521526

522-
resource_config = {"current_host": host, "hosts": self.hosts}
527+
resource_config = {
528+
"current_host": host,
529+
"hosts": self.hosts,
530+
"network_interface_name": "eth0",
531+
"current_instance_type": self.instance_type,
532+
}
523533

524534
json_input_data_config = {}
525535
for c in input_data_config:

0 commit comments

Comments
 (0)