Merge 53810a5 into 72374ef

Author: DaltheCow

pyproject.toml

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

src/guidellm/benchmark/output.py

@@ -1,3 +1,4 @@
+from copy import deepcopy
 import csv
 import json
 import math
@@ -6,7 +7,6 @@
 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",
@@ -236,14 +238,14 @@ def save_html(self, path: Union[str, Path]) -> Path:
         :param path: The path to create the report at.
         :return: The path to the report.
         """
-
+        
         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/utils/__init__.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/dict.py

@@ -0,0 +1,21 @@
+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 key, value in d.items():
+    recursive_key_update(value, key_update_func)
+  return d

src/guidellm/utils/text.py

@@ -14,6 +14,7 @@
 
 __all__ = [
     "EndlessTextCreator",
+    "camelize_str",
     "clean_text",
     "filter_text",
     "is_puncutation",
@@ -189,6 +190,10 @@ 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,
@@ -213,4 +218,4 @@ def create_text(self, start: int, length: int) -> str:
 
             text += add_word
 
-        return text
+        return text

src/ui/app/layout.tsx

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

src/ui/lib/store/runInfoWindowData.ts

@@ -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/slices/runInfo/runInfo.api.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/workloadDetails/workloadDetails.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({

src/ui/lib/store/workloadDetailsWindowData.ts

@@ -1,4 +1,4 @@
-export const workloadDetailsScript = `window.workload_details = {
+export const workloadDetailsScript = `window.workloadDetails = {
   "prompts": {
     "samples": [
       "such a sacrifice to her advantage as years of gratitude cannot enough acknowledge. By this time she is actually with them! If such goodness does not make her miserable now, she will never deserve to be happy! What a meeting for her, when she first sees my aunt! We must endeavour to forget all that has passed on either side, said Jane I hope and trust they will yet be happy. His consenting to marry her is a proof, I will believe, that he is come to a right way of thinking. Their mutual affection will steady them; and I flatter myself they will settle so quietly, and live in so rational a manner",