From 0f71906fb062b18e9a4efd2240c1e50193182bf4 Mon Sep 17 00:00:00 2001
From: Jordan Singer <jordan.singer@klo.dev>
Date: Tue, 21 Nov 2023 11:42:12 -0500
Subject: [PATCH] fix for lb viz

---
 pkg/templates/aws/edges/load_balancer-subnet.yaml     | 4 +++-
 pkg/templates/aws/edges/target_group-ecs_service.yaml | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/pkg/templates/aws/edges/load_balancer-subnet.yaml b/pkg/templates/aws/edges/load_balancer-subnet.yaml
index de5ad8006..01e111473 100644
--- a/pkg/templates/aws/edges/load_balancer-subnet.yaml
+++ b/pkg/templates/aws/edges/load_balancer-subnet.yaml
@@ -1,4 +1,6 @@
 source: aws:load_balancer
 target: aws:subnet
 
-edge_weight_multiplier: 200
\ No newline at end of file
+edge_weight_multiplier: 200
+classification:
+  - target_group
\ No newline at end of file
diff --git a/pkg/templates/aws/edges/target_group-ecs_service.yaml b/pkg/templates/aws/edges/target_group-ecs_service.yaml
index f29f408b2..9a2769e8e 100644
--- a/pkg/templates/aws/edges/target_group-ecs_service.yaml
+++ b/pkg/templates/aws/edges/target_group-ecs_service.yaml
@@ -14,3 +14,5 @@ operational_rules:
         configuration:
           field: TargetType
           value: ip
+classification:
+  - service_endpoint
\ No newline at end of file