BUG: update fast_scalar_power to handle special-case squaring for any array type except object arrays

MaanasArora · MaanasArora · commit 81f9adf35707 · 2025-07-16T18:11:31.000-04:00
diff --git a/numpy/_core/src/multiarray/number.c b/numpy/_core/src/multiarray/number.c
@@ -363,7 +363,12 @@ fast_scalar_power(PyObject *o1, PyObject *o2, int inplace, PyObject **result)
     }
 
     PyArrayObject *a1 = (PyArrayObject *)o1;
-    if (!(PyArray_ISFLOAT(a1) || PyArray_ISCOMPLEX(a1))) {
+    if (PyArray_ISOBJECT(a1)) {
+        return 1;
+    }
+    if (!is_square && !PyArray_ISFLOAT(a1) && !PyArray_ISCOMPLEX(a1)) {
+        // we special-case squaring for any array type
+        // gh-29388
         return 1;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -363,7 +363,12 @@ fast_scalar_power(PyObject o1, PyObject o2, int inplace, PyObject **result)`
`363`	`363`	`}`
`364`	`364`
`365`	`365`	`PyArrayObject a1 = (PyArrayObject )o1;`
`366`		`- if (!(PyArray_ISFLOAT(a1) \|\| PyArray_ISCOMPLEX(a1))) {`
	`366`	`+ if (PyArray_ISOBJECT(a1)) {`
	`367`	`+ return 1;`
	`368`	`+ }`
	`369`	`+ if (!is_square && !PyArray_ISFLOAT(a1) && !PyArray_ISCOMPLEX(a1)) {`
	`370`	`+ // we special-case squaring for any array type`
	`371`	`+ // gh-29388`
`367`	`372`	`return 1;`
`368`	`373`	`}`
`369`	`374`