From eeecedbdf93485e2e123a3ed2c697f2892981ef1 Mon Sep 17 00:00:00 2001
From: c-bata <contact@c-bata.link>
Date: Sun, 4 Apr 2021 15:07:08 +0900
Subject: [PATCH] Reduce the size of attr values

---
 optuna_dashboard/serializer.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/optuna_dashboard/serializer.py b/optuna_dashboard/serializer.py
index e164ff78..4959b7ac 100644
--- a/optuna_dashboard/serializer.py
+++ b/optuna_dashboard/serializer.py
@@ -1,4 +1,3 @@
-import json
 from typing import Any, Dict, List, Tuple
 
 from optuna.distributions import BaseDistribution
@@ -11,6 +10,7 @@ except ImportError:
     from typing_extensions import TypedDict
 
 
+MAX_ATTR_LENGTH = 128
 Attribute = TypedDict(
     "Attribute",
     {
@@ -35,7 +35,21 @@ TrialParam = TypedDict(
 
 
 def serialize_attrs(attrs: Dict[str, Any]) -> List[Attribute]:
-    return [{"key": k, "value": json.dumps(v)} for k, v in attrs.items()]
+    serialized = []
+    for k, v in attrs.items():
+        value: str
+        if isinstance(v, str):
+            value = v[:MAX_ATTR_LENGTH] if len(v) > MAX_ATTR_LENGTH else v
+        elif isinstance(v, (bool, float, int)):
+            value = str(v)
+        elif isinstance(v, bytes):
+            value = "<binary object>"
+        elif v is None:
+            value = "None"
+        else:  # unsupported type
+            continue
+        serialized.append({"key": k, "value": value})
+    return serialized
 
 
 def serialize_intermediate_values(values: Dict[int, float]) -> List[IntermediateValue]: