reintroducing the service labels that were removed earlier by mistake

SebastianScherer88 · Oct 29, 2024 · 2fd1a5e · 2fd1a5e
1 parent fe1c262
commit 2fd1a5e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sdk/bettmensch_ai/components/torch_utils.py b/sdk/bettmensch_ai/components/torch_utils.py
@@ -152,8 +152,11 @@ def create_torch_ddp_service_template(
         manifest=f"""apiVersion: v1
 kind: Service
 metadata:
-  name: {component_task_name}-{{{{workflow.name}}}}
+  name: {component_task_name}-{{{{workflow.uid}}}}
   namespace: {namespace}
+  labels:
+    workflows.argoproj.io/workflow: {{{{workflow.name}}}}
+    torch-job: {component_task_name}
 spec:
   clusterIP: None  # ClusterIP set to None for headless service.
   ports: