Skip to content

Commit 7885469

Browse files
committed
fix: Add missing attributes to local resourceconfig
1 parent 0402ba0 commit 7885469

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/sagemaker/local/image.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -519,7 +519,12 @@ def write_config_files(self, host, hyperparameters, input_data_config):
519519
"""
520520
config_path = os.path.join(self.container_root, host, "input", "config")
521521

522-
resource_config = {"current_host": host, "hosts": self.hosts}
522+
resource_config = {
523+
"current_host": host,
524+
"hosts": self.hosts,
525+
"network_interface_name": "eth0",
526+
"current_instance_type": self.instance_type,
527+
}
523528

524529
json_input_data_config = {}
525530
for c in input_data_config:

0 commit comments

Comments
 (0)