We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents c6028fb + 96032d8 commit eec7a47Copy full SHA for eec7a47
1 file changed
src/blosc2/lazyexpr.py
@@ -1673,7 +1673,7 @@ def slices_eval_getitem(
1673
def infer_reduction_dtype(dtype, operation):
1674
# It may change in the future, but mostly array-api compliant
1675
my_float = np.result_type(
1676
- dtype, np.float32 if dtype == np.float32 or dtype == np.complex64 else blosc2.DEFAULT_FLOAT
+ dtype, np.float32 if dtype in (np.float32, np.complex64) else blosc2.DEFAULT_FLOAT
1677
)
1678
if operation in {ReduceOp.SUM, ReduceOp.PROD}:
1679
if np.issubdtype(dtype, np.unsignedinteger):
0 commit comments