Unique worker group names and v0.1.5 bump

Maxusmusti · Maxusmusti · commit 3d8ce269c9f5 · 2022-11-03T23:47:50.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codeflare-sdk"
-version = "0.1.4"
+version = "0.1.5"
 description = "Python SDK for codeflare client"
 
 license = "GPL-3.0-only"
diff --git a/src/codeflare_sdk/utils/generate_yaml.py b/src/codeflare_sdk/utils/generate_yaml.py
@@ -122,6 +122,7 @@ def update_nodes(item, appwrapper_name, min_cpu, max_cpu, min_memory, max_memory
         worker["replicas"] = workers
         worker["minReplicas"] = workers
         worker["maxReplicas"] = workers
+        worker["groupName"] = "small-group-" + appwrapper_name
         worker["rayStartParams"]["num-gpus"] = str(int(gpu))
 
         for comp in [head, worker]:
