impy-array 2.4.1__tar.gz → 2.4.2__tar.gz

{impy_array-2.4.1 → impy_array-2.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: impy-array
-Version: 2.4.1
+Version: 2.4.2
 Summary: Speed up coding/extending image analysis in Python.
 Project-URL: Download, https://github.com/hanjinliu/impy
 Author-email: Hanjin Liu <liuhanjin-sc@g.ecc.u-tokyo.ac.jp>

{impy_array-2.4.1 → impy_array-2.4.2}/impy/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.4.1"
+__version__ = "2.4.2"
 __author__ = "Hanjin Liu"
 __email__ = "liuhanjin-sc@g.ecc.u-tokyo.ac.jp"
 

{impy_array-2.4.1 → impy_array-2.4.2}/impy/arrays/_utils/_glcm.py

@@ -130,4 +130,4 @@ def check_glcm(self, bins, rescale_max):
         raise ValueError(f"`bins` must be a divisor of {imax+1} (max value of {self.dtype}).")
     self = self // ((imax+1) // bins)
 
-    return self, bins, rescale_max
+    return self, bins, rescale_max

{impy_array-2.4.1 → impy_array-2.4.2}/impy/arrays/_utils/_structures.py

@@ -37,4 +37,4 @@ def ball_like_odd(radius, ndim):
     l = xc*2 + 1
     coords = xp.indices((l,)*ndim)
     s = xp.sum((a-xc)**2 for a in coords)
-    return xp.array(s <= radius*radius, dtype=bool)
+    return xp.array(s <= radius*radius, dtype=bool)

{impy_array-2.4.1 → impy_array-2.4.2}/impy/arrays/lazy.py

@@ -281,6 +281,12 @@ class LazyImgArray(AxesMixin):
         chunk_info = ", ".join([f"{s}({o})" for s, o in zip(self.chunksize, self.axes)])
         return chunk_info
     
+    def astype(self, dtype: DTypeLike, copy: bool = True) -> LazyImgArray:
+        """Convert the array to the specified data type."""
+        out = self.__class__(self.value.astype(dtype, copy=copy))
+        out._set_info(self)
+        return out
+    
     def _repr_dict_(self):
         return {
             "name": self.name,
@@ -555,9 +561,9 @@ class LazyImgArray(AxesMixin):
             dtype = self.dtype
         all_axes = str(self.axes)
         def _func(input: xp.ndarray, *args, **kwargs):
-            out = xp.empty(input.shape, input.dtype)
+            out = xp.empty(input.shape, dtype=dtype)
             for sl in iter_slice(input.shape, c_axes, all_axes):
-                out[sl] = func(input[sl], *args, **kwargs)
+                out[sl] = func(input[sl], *args, **kwargs).astype(dtype, copy=False)
             return out
         depth = switch_slice(c_axes, self.axes, 0, depth)
         return da.map_overlap(

{impy_array-2.4.1 → impy_array-2.4.2}/impy/arrays/tiled.py

@@ -94,9 +94,12 @@ class TiledImage(Generic[_T]):
             from dask import array as da
             
             input = da.from_array(img.value, chunks=self._chunks)
+            
+            def _func(arr: np.ndarray, *args, **kwargs) -> np.ndarray:
+                return func(arr, *args, **kwargs).astype(arr.dtype, copy=False)
             out: np.ndarray = xp.asnumpy(
                 da.map_overlap(
-                    func, 
+                    _func, 
                     input,
                     *args,
                     depth=self.depth,
@@ -114,6 +117,7 @@ class TiledImage(Generic[_T]):
                 c_axes="",
                 depth=self.depth,
                 boundary=self.boundary, 
+                dtype=img.dtype,
                 args=args,
                 kwargs=kwargs,
             )

{impy_array-2.4.1 → impy_array-2.4.2}/impy/core.py

@@ -121,8 +121,11 @@ def array(
             dtype = arr.dtype
     
     axes, name = _normalize_params(axes, name, like)
-        
-    _arr = np.array(arr, dtype=dtype, copy=copy)
+    
+    if copy:
+        _arr = np.array(arr, dtype=dtype)
+    else:
+        _arr = np.asarray(arr, dtype=dtype)
     
     # Automatically determine axes
     if axes is None:

{impy_array-2.4.1 → impy_array-2.4.2}/impy/utils/gauss.py

@@ -156,4 +156,4 @@ class GaussianBackground(DiagonalGaussian):
         self.sg = np.array(data.shape, dtype="float32")
         self.b, p95 = np.percentile(data, [5, 95])
         self.a = p95 - self.b
-        return None
+        return None