From 439c0e94cc49bcce67f940da44f0033791dc137a Mon Sep 17 00:00:00 2001
From: Fiddle-Config Team <fiddle-config-team@google.com>
Date: Tue, 7 Jan 2025 17:53:52 -0800
Subject: [PATCH] Add type annotation for bytes and bytearray for
 `serialization.load_json` `serialized_value` argument

PiperOrigin-RevId: 713102772
---
 fiddle/_src/experimental/serialization.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/fiddle/_src/experimental/serialization.py b/fiddle/_src/experimental/serialization.py
index 55113600..3bc8bf65 100644
--- a/fiddle/_src/experimental/serialization.py
+++ b/fiddle/_src/experimental/serialization.py
@@ -35,7 +35,7 @@
 import re
 import sys
 import types
-from typing import Any, Dict, Iterable, List, Optional, Type
+from typing import Any, Dict, Iterable, List, Optional, Type, Union
 
 from fiddle import daglish
 from fiddle._src import config as config_lib
@@ -844,7 +844,7 @@ def dump_json(
 
 
 def load_json(
-    serialized_value: str,
+    serialized_value: Union[str, bytes, bytearray],
     pyref_policy: Optional[PyrefPolicy] = None,
 ) -> Any:
   """Returns a Python object deserialized from `serialized_value`.