fix unstack

lucascolley · lucascolley · commit e5292844381c · 2025-01-01T17:44:05.000Z
diff --git a/src/pint_array/__init__.py b/src/pint_array/__init__.py
@@ -315,7 +315,7 @@ def manip_fun(x, *args, **kwargs):
             else:
                 magnitude = xp_func(magnitude, *args, **kwargs)
 
-            if name in output_arrays:
+            if func_str in output_arrays:
                 return tuple(
                     ArrayUnitQuantity(magnitude_i, units) for magnitude_i in magnitude
                 )

Original file line number	Diff line number	Diff line change
`@@ -315,7 +315,7 @@ def manip_fun(x, args, *kwargs):`
`315`	`315`	`else:`
`316`	`316`	`magnitude = xp_func(magnitude, args, *kwargs)`
`317`	`317`
`318`		`- if name in output_arrays:`
	`318`	`+ if func_str in output_arrays:`
`319`	`319`	`return tuple(`
`320`	`320`	`ArrayUnitQuantity(magnitude_i, units) for magnitude_i in magnitude`
`321`	`321`	`)`