Added to dict function to account for get cluster fails

Signed-off-by: Pat O'Connor <[email protected]>

Author: chipspeak

src/codeflare_sdk/ray/cluster/cluster.py

@@ -21,7 +21,6 @@
 from time import sleep
 from typing import List, Optional, Tuple, Dict
 import copy
-import dataclasses
 
 from ray.job_submission import JobSubmissionClient, JobStatus
 import time
@@ -1060,9 +1059,7 @@ def get_cluster(
     )
     # 1. Prepare RayClusterSpec from ClusterConfiguration
     # Create a temporary config with appwrapper=False to ensure build_ray_cluster returns RayCluster YAML
-    temp_cluster_config_dict = dataclasses.asdict(cluster_config)
-    # Filter out None values similar to exclude_none=True in Pydantic
-    temp_cluster_config_dict = {k: v for k, v in temp_cluster_config_dict.items() if v is not None}
+    temp_cluster_config_dict = cluster_config.dict(exclude_none=True)
     temp_cluster_config_dict["appwrapper"] = False
     # Set overwrite_default_resource_mapping=True to avoid conflicts when extended_resource_mapping 
     # already contains the combined default and custom mappings from the original object

src/codeflare_sdk/ray/cluster/config.py

@@ -289,3 +289,19 @@ def check_type(value, expected_type):
             return isinstance(value, expected_type)
 
         return check_type(value, expected_type)
+
+    def dict(self, exclude_none: bool = False):
+        """
+        Convert the ClusterConfiguration to a dictionary.
+        
+        Args:
+            exclude_none (bool): If True, exclude fields with None values.
+            
+        Returns:
+            dict: Dictionary representation of the ClusterConfiguration.
+        """
+        import dataclasses
+        result = dataclasses.asdict(self)
+        if exclude_none:
+            result = {k: v for k, v in result.items() if v is not None}
+        return result