From 78854691cd78115dae7d709591207a7a8f7b5bb9 Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Tue, 21 Jan 2025 21:10:49 +0000 Subject: [PATCH 1/4] fix: Add missing attributes to local resourceconfig --- src/sagemaker/local/image.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/sagemaker/local/image.py b/src/sagemaker/local/image.py index ef24bb0d99..d76d6af9f3 100644 --- a/src/sagemaker/local/image.py +++ b/src/sagemaker/local/image.py @@ -519,7 +519,12 @@ def write_config_files(self, host, hyperparameters, input_data_config): """ config_path = os.path.join(self.container_root, host, "input", "config") - resource_config = {"current_host": host, "hosts": self.hosts} + resource_config = { + "current_host": host, + "hosts": self.hosts, + "network_interface_name": "eth0", + "current_instance_type": self.instance_type, + } json_input_data_config = {} for c in input_data_config: From 759b0c23de6c92b87a7c758b31daafb1ce13df54 Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Tue, 21 Jan 2025 22:40:10 +0000 Subject: [PATCH 2/4] format fix --- src/sagemaker/local/image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/local/image.py b/src/sagemaker/local/image.py index d76d6af9f3..0782d824c4 100644 --- a/src/sagemaker/local/image.py +++ b/src/sagemaker/local/image.py @@ -520,8 +520,8 @@ def write_config_files(self, host, hyperparameters, input_data_config): config_path = os.path.join(self.container_root, host, "input", "config") resource_config = { - "current_host": host, - "hosts": self.hosts, + "current_host": host, + "hosts": self.hosts, "network_interface_name": "eth0", "current_instance_type": self.instance_type, } From 84330cfe84e59e799ef2a79fadc748b53148a2b1 Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Tue, 21 Jan 2025 23:11:08 +0000 Subject: [PATCH 3/4] add missing for local processing --- src/sagemaker/local/image.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/sagemaker/local/image.py b/src/sagemaker/local/image.py index 0782d824c4..36916a6a53 100644 --- a/src/sagemaker/local/image.py +++ b/src/sagemaker/local/image.py @@ -473,7 +473,12 @@ def write_processing_config_files( """ config_path = os.path.join(self.container_root, host, "config") - resource_config = {"current_host": host, "hosts": self.hosts} + resource_config = { + "current_host": host, + "hosts": self.hosts, + "network_interface_name": "eth0", + "current_instance_type": self.instance_type, + } _write_json_file(os.path.join(config_path, "resourceconfig.json"), resource_config) processing_job_config = { From c1d9a4de161c9ef2a8cc981b0d022efd528679dd Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Tue, 21 Jan 2025 23:16:58 +0000 Subject: [PATCH 4/4] format fix --- src/sagemaker/local/image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/local/image.py b/src/sagemaker/local/image.py index 36916a6a53..3d0f8394ab 100644 --- a/src/sagemaker/local/image.py +++ b/src/sagemaker/local/image.py @@ -474,7 +474,7 @@ def write_processing_config_files( config_path = os.path.join(self.container_root, host, "config") resource_config = { - "current_host": host, + "current_host": host, "hosts": self.hosts, "network_interface_name": "eth0", "current_instance_type": self.instance_type,