Skip to content

Commit a0c39ef

Browse files
authored
fix: Add missing attributes to local resourceconfig (#4999)
* fix: Add missing attributes to local resourceconfig * format fix * add missing for local processing * format fix
1 parent 87554bb commit a0c39ef

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

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)