Skip to content

Commit a428dbe

Browse files
committed
Use make_function_def in deserializer_generation.py
1 parent e0d8a36 commit a428dbe

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/lionweb/generation/deserializer_generation.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
import astor # type: ignore
77

8+
from lionweb.generation.utils import make_function_def
89
from lionweb.language import Concept, Language
910

1011

@@ -55,7 +56,7 @@ def deserializer_generation(click, language: Language, output):
5556
concept_name = cast(str, language_element.get_name())
5657
# deserializer() inner function
5758
register_func_body.append(
58-
ast.FunctionDef(
59+
make_function_def(
5960
name=f"deserializer_{concept_name.lower()}",
6061
args=ast.arguments(
6162
posonlyargs=[],
@@ -135,7 +136,7 @@ def deserializer_generation(click, language: Language, output):
135136
)
136137
)
137138

138-
register_func = ast.FunctionDef(
139+
register_func = make_function_def(
139140
name="register_deserializers",
140141
args=ast.arguments(
141142
posonlyargs=[],

0 commit comments

Comments
 (0)