add custom dict camelize logic (#246)

This removes reliance on the library pyhumps converting dicts keys from
snake_case to camelCase for use in the UI

---------

Signed-off-by: dalthecow <dalcowboiz@gmail.com>
Signed-off-by: Samuel Monson <smonson@redhat.com>
Co-authored-by: Samuel Monson <smonson@redhat.com>

Author: DaltheCow

pylock.toml

@@ -56,7 +56,6 @@ dependencies = [
     "protobuf",
     "pydantic>=2.11.7",
     "pydantic-settings>=2.0.0",
-    "pyhumps>=3.8.0",
     "pyyaml>=6.0.0",
     "rich",
     "transformers",

pyproject.toml

@@ -2,11 +2,11 @@
 import json
 import math
 from collections import OrderedDict
+from copy import deepcopy
 from datetime import datetime
 from pathlib import Path
 from typing import Any, Literal, Optional, Union
 
-import humps  # type: ignore[import-not-found]
 import yaml
 from pydantic import Field
 from rich.console import Console
@@ -30,6 +30,8 @@
 from guidellm.presentation.injector import create_report
 from guidellm.scheduler import strategy_display_str
 from guidellm.utils import Colors, split_text_list_by_length
+from guidellm.utils.dict import recursive_key_update
+from guidellm.utils.text import camelize_str
 
 __all__ = [
     "GenerativeBenchmarksConsole",
@@ -239,11 +241,11 @@ def save_html(self, path: Union[str, Path]) -> Path:
 
         data_builder = UIDataBuilder(self.benchmarks)
         data = data_builder.to_dict()
-        camel_data = humps.camelize(data)
+        camel_data = recursive_key_update(deepcopy(data), camelize_str)
         ui_api_data = {}
         for k, v in camel_data.items():
-            key = f"window.{humps.decamelize(k)} = {{}};"
-            value = f"window.{humps.decamelize(k)} = {json.dumps(v, indent=2)};\n"
+            key = f"window.{k} = {{}};"
+            value = f"window.{k} = {json.dumps(v, indent=2)};\n"
             ui_api_data[key] = value
         return create_report(ui_api_data, path)
 

src/guidellm/benchmark/output.py

@@ -1,5 +1,6 @@
 from .colors import Colors
 from .default_group import DefaultGroupHandler
+from .dict import recursive_key_update
 from .hf_datasets import (
     SUPPORTED_TYPES,
     save_dataset_to_file,
@@ -10,6 +11,7 @@
 from .random import IntegerRangeSampler
 from .text import (
     EndlessTextCreator,
+    camelize_str,
     clean_text,
     filter_text,
     is_puncutation,
@@ -24,11 +26,13 @@
     "DefaultGroupHandler",
     "EndlessTextCreator",
     "IntegerRangeSampler",
+    "camelize_str",
     "check_load_processor",
     "clean_text",
     "filter_text",
     "is_puncutation",
     "load_text",
+    "recursive_key_update",
     "save_dataset_to_file",
     "split_text",
     "split_text_list_by_length",

src/guidellm/utils/__init__.py

@@ -0,0 +1,23 @@
+def recursive_key_update(d, key_update_func):
+    if not isinstance(d, dict) and not isinstance(d, list):
+        return d
+
+    if isinstance(d, list):
+        for item in d:
+            recursive_key_update(item, key_update_func)
+        return d
+
+    updated_key_pairs = []
+    for key, _ in d.items():
+        updated_key = key_update_func(key)
+        if key != updated_key:
+            updated_key_pairs.append((key, updated_key))
+
+    for key_pair in updated_key_pairs:
+        old_key, updated_key = key_pair
+        d[updated_key] = d[old_key]
+        del d[old_key]
+
+    for _, value in d.items():
+        recursive_key_update(value, key_update_func)
+    return d

src/guidellm/utils/dict.py

@@ -14,6 +14,7 @@
 
 __all__ = [
     "EndlessTextCreator",
+    "camelize_str",
     "clean_text",
     "filter_text",
     "is_puncutation",
@@ -189,6 +190,12 @@ def is_puncutation(text: str) -> bool:
     return len(text) == 1 and not text.isalnum() and not text.isspace()
 
 
+def camelize_str(snake_case_string: str) -> str:
+    return (words := snake_case_string.split("_"))[0].lower() + "".join(
+        word.capitalize() for word in words[1:]
+    )
+
+
 class EndlessTextCreator:
     def __init__(
         self,

src/guidellm/utils/text.py

@@ -34,7 +34,7 @@ export default function RootLayout({
     <script
       dangerouslySetInnerHTML={{
         __html:
-          'window.run_info = {}; window.workload_details = {}; window.benchmarks = {};',
+          'window.runInfo = {}; window.workloadDetails = {}; window.benchmarks = {};',
       }}
     />
   );

src/ui/app/layout.tsx

@@ -1,4 +1,4 @@
-export const runInfoScript = `window.run_info = {
+export const runInfoScript = `window.runInfo = {
   "model": {
     "name": "neuralmagic/Qwen2.5-7B-quantized.w8a8",
     "size": 0

src/ui/lib/store/runInfoWindowData.ts

@@ -5,7 +5,7 @@ import { RunInfo } from './runInfo.interfaces';
 const USE_MOCK_API = process.env.NEXT_PUBLIC_USE_MOCK_API === 'true';
 
 const fetchRunInfo = () => {
-  return { data: window.run_info as RunInfo };
+  return { data: window.runInfo as RunInfo };
 };
 
 export const runInfoApi = createApi({

src/ui/lib/store/slices/runInfo/runInfo.api.ts

@@ -5,7 +5,7 @@ import { WorkloadDetails } from './workloadDetails.interfaces';
 const USE_MOCK_API = process.env.NEXT_PUBLIC_USE_MOCK_API === 'true';
 
 const fetchWorkloadDetails = () => {
-  return { data: window.workload_details as WorkloadDetails };
+  return { data: window.workloadDetails as WorkloadDetails };
 };
 
 export const workloadDetailsApi = createApi({