From 3d14a3d46fbf5df8a589e9a9b2a78ef024faff84 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Wed, 22 Jan 2025 22:54:13 -0800
Subject: [PATCH] Kill colons

---
 tests/client-sdk/report.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/tests/client-sdk/report.py b/tests/client-sdk/report.py
index de50efa46..cf7a84d7f 100644
--- a/tests/client-sdk/report.py
+++ b/tests/client-sdk/report.py
@@ -147,7 +147,7 @@ class Report:
     def pytest_sessionfinish(self, session):
         report = []
         report.append(f"# Report for {self.image_name} distribution")
-        report.append("\n## Supported Models:")
+        report.append("\n## Supported Models")
 
         header = f"| Model Descriptor | {self.image_name} |"
         dividor = "|:---|:---|"
@@ -180,7 +180,7 @@ class Report:
                 rows.append(row)
         report.extend(rows)
 
-        report.append("\n## Inference:")
+        report.append("\n## Inference")
         test_table = [
             "| Model | API | Capability | Test | Status |",
             "|:----- |:-----|:-----|:-----|:-----|",
@@ -205,9 +205,10 @@ class Report:
 
         report.extend(test_table)
 
+        name_map = {Api.vector_io: "Vector IO", Api.agents: "Agents"}
         for api_group in [Api.vector_io, Api.agents]:
-            api_capitalized = api_group.name.capitalize()
-            report.append(f"\n## {api_capitalized}:")
+            api_capitalized = name_map[api_group]
+            report.append(f"\n## {api_capitalized}")
             test_table = [
                 "| API | Capability | Test | Status |",
                 "|:-----|:-----|:-----|:-----|",