fix lib naming bugs

Signed-off-by: HH <hhcs9527@gmail.com>
flyteorg · Jul 25, 2023 · 427e807 · 427e807
1 parent fcab254
commit 427e807
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flytekit/core/type_engine.py b/flytekit/core/type_engine.py
@@ -352,7 +352,7 @@ def to_literal(self, ctx: FlyteContext, python_val: T, python_type: Type[T], exp
                 f"Dataclass {python_type} should be decorated with @dataclass_json to be " f"serialized correctly"
             )
         self._serialize_flyte_type(python_val, python_type)
-        cb = CodeBuilder(python_type, format_name="json", encoder=json.dumps, decoder=json.loads) # type: ignore
+        cb = CodeBuilder(python_type, format_name="json", encoder=_json.dumps, decoder=_json.loads) # type: ignore
         cb.add_pack_method()
 
         return Literal(
@@ -607,7 +607,7 @@ def to_python_value(self, ctx: FlyteContext, lv: Literal, expected_python_type:
                 f"serialized correctly"
             )
         json_str = _json_format.MessageToJson(lv.scalar.generic)
-        cb = CodeBuilder(expected_python_type, format_name="json", encoder=json.dumps, decoder=json.loads) # type: ignore
+        cb = CodeBuilder(expected_python_type, format_name="json", encoder=_json.dumps, decoder=_json.loads) # type: ignore
         cb.add_unpack_method()
 
         dc = cast(DataClassJsonMixin, expected_python_type).from_json(json_str)  # type: ignore