diff --git a/src/codeflare_sdk/ray/cluster/config.py b/src/codeflare_sdk/ray/cluster/config.py index dc61de2a..803acb4d 100644 --- a/src/codeflare_sdk/ray/cluster/config.py +++ b/src/codeflare_sdk/ray/cluster/config.py @@ -200,6 +200,10 @@ def _validate_extended_resource_requests(self, extended_resources: Dict[str, int f"extended resource '{k}' not found in extended_resource_mapping, available resources are {list(self.extended_resource_mapping.keys())}, to add more supported resources use extended_resource_mapping. i.e. extended_resource_mapping = {{'{k}': 'FOO_BAR'}}" ) + # add mock method to trigger e2es + def mock_method(self): + return "mock_method" + def _str_mem_no_unit_add_GB(self): if ( isinstance(self.worker_memory_requests, str)