Group all imread functions together in the same file (#290)

dask · Mar 14, 2023 · 96f2795 · 96f2795
1 parent 69989c7
commit 96f2795
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/dask_image/imread/__init__.py b/dask_image/imread/__init__.py
@@ -8,8 +8,6 @@
 import pims
 from tifffile import natural_sorted
 
-from . import _utils
-
 
 def imread(fname, nframes=1, *, arraytype="numpy"):
     """
@@ -98,4 +96,9 @@ def _map_read_frame(x, multiple_files, block_info=None, **kwargs):
     else:
         i, j = block_info[None]['array-location'][0]
 
-    return _utils._read_frame(fn=fn, i=slice(i, j), **kwargs)
+    return _read_frame(fn=fn, i=slice(i, j), **kwargs)
+
+
+def _read_frame(fn, i, *, arrayfunc=np.asanyarray):
+    with pims.open(fn) as imgs:
+        return arrayfunc(imgs[i])
diff --git a/dask_image/imread/_utils.py b/dask_image/imread/_utils.py