Skip to content

Commit 81d368e

Browse files
committed
only generate add fun upon multiple
1 parent ce7325c commit 81d368e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

easyDataverse/classgen.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -122,10 +122,12 @@ def create_compound(
122122
# Set non-multiple compounds directly as default
123123
field_meta.default = None
124124
field_meta.default_factory = sub_cls
125-
126-
# Create add function
127-
fun_name = f"add_{attribute_name}"
128-
add_functions[fun_name] = generate_add_function(sub_cls, attribute_name, fun_name)
125+
else:
126+
# Create add function
127+
fun_name = f"add_{attribute_name}"
128+
add_functions[fun_name] = generate_add_function(
129+
sub_cls, attribute_name, fun_name
130+
)
129131

130132
return (dtype, field_meta)
131133

0 commit comments

Comments
 (0)