diff --git a/bsplines2d/_class02_operators.py b/bsplines2d/_class02_operators.py index 21657b1..4fc786f 100644 --- a/bsplines2d/_class02_operators.py +++ b/bsplines2d/_class02_operators.py @@ -525,6 +525,7 @@ def apply_operator( keybs=keybs, # for units operator=operator, + integ_op=integ_op, ) # ----------------- @@ -691,6 +692,7 @@ def _apply_operator_prepare( key=None, keybs=None, operator=None, + integ_op=None, ): # ---------- @@ -708,14 +710,6 @@ def _apply_operator_prepare( refbs = coll.dobj[wbs][keybs]['ref'] - # ---------- - # unitsbs - - unitsbs = [ - coll.ddata[k0]['units'] - for k0 in coll.dobj[wbs][keybs]['apex'] - ] - # -------------- # fill dict @@ -745,9 +739,7 @@ def _apply_operator_prepare( # units units0 = asunits.Unit(coll.ddata[k0]['units']) if operator == 'D0N1': - units = units0 - for uu in unitsbs: - units = units * uu + units = units0 * integ_op['M']['units'] else: raise NotImplementedError()