update head node name on relevant modules

Author: Cohen-J-Omer

src/ibm_ray_config/modules/gen2/ray/floating_ip.py

@@ -24,7 +24,7 @@ def run(self) -> Dict[str, Any]:
                     else:
                         self.base_config['available_node_types'][available_node_type]['node_config'].pop('head_ip', None)
             else:
-                self.base_config['available_node_types']['ray_head_default']['node_config']['head_ip'] = head_ip
+                self.base_config['available_node_types'][self.DEFAULT_NODE_TYPE]['node_config']['head_ip'] = head_ip
 
         return self.base_config
 

src/ibm_ray_config/modules/gen2/ray/image.py

@@ -18,5 +18,5 @@ def update_config(self, image_id, minimum_provisioned_size, custom_image):
                 self.base_config['available_node_types'][available_node_type]['node_config']['image_id'] = image_id
                 self.base_config['available_node_types'][available_node_type]['node_config']['boot_volume_capacity'] = minimum_provisioned_size
         else:
-            self.base_config['available_node_types']['ray_head_default']['node_config']['image_id'] = image_id
-            self.base_config['available_node_types']['ray_head_default']['node_config']['boot_volume_capacity'] = minimum_provisioned_size
+            self.base_config['available_node_types'][self.DEFAULT_NODE_TYPE]['node_config']['image_id'] = image_id
+            self.base_config['available_node_types'][self.DEFAULT_NODE_TYPE]['node_config']['boot_volume_capacity'] = minimum_provisioned_size

src/ibm_ray_config/modules/gen2/ray/ssh_key.py

@@ -21,4 +21,4 @@ def update_config(self, ssh_key_id, ssh_key_path, ssh_user):
             for available_node_type in self.base_config['available_node_types']:
                 self.base_config['available_node_types'][available_node_type]['node_config']['key_id'] = ssh_key_id
         else:
-            self.base_config['available_node_types'] = {'ray_head_default': {'node_config': {'key_id': ssh_key_id}}}
+            self.base_config['available_node_types'] = {self.DEFAULT_NODE_TYPE: {'node_config': {'key_id': ssh_key_id}}}