Manually cleanup some very nested formatting.
@@ -333,12 +333,8 @@ def _untuple_broadcast_only_before_after(before, after):
after_param_ref = building_blocks.Reference(
after_param_name, after_param_type
)
- untupled_after = building_blocks.Lambda(
- after_param_name,
- after_param_type,
- building_blocks.Call(
- after,
- building_blocks.Struct([
+
+ after_result_arg = building_blocks.Struct([
(
'original_arg',
building_blocks.Selection(after_param_ref, 'original_arg'),
@@ -352,8 +348,15 @@ def _untuple_broadcast_only_before_after(before, after):
),
)]),
),
- ]),
- ),
+ ])
+ after_result = building_blocks.Call(
+ after,
+ after_result_arg,
+ )
+ untupled_after = building_blocks.Lambda(
+ after_param_name,
+ after_param_type,
+ after_result,
)
return untupled_before, untupled_after
