Skip to content

Commit

Permalink
remove image loader changes
Browse files Browse the repository at this point in the history
  • Loading branch information
Benjamin Morris committed Jul 1, 2024
1 parent 9e84b76 commit 7ce9946
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions cyto_dl/image/io/aicsimage_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,12 +49,6 @@ def __init__(
self.dtype = dtype
self.dask_load = dask_load

def split_args(self, arg):
if ',' in str(arg):
return list(map(int, arg.split(",")))
else:
return arg

def __call__(self, data):
# copying prevents the dataset from being modified inplace - important when using partially cached datasets so that the memory use doesn't increase over time
data = data.copy()
Expand All @@ -64,7 +58,7 @@ def __call__(self, data):
img = BioImage(path)
if self.scene_key in data:
img.set_scene(data[self.scene_key])
kwargs = {k: self.split_args(data[k]) for k in self.kwargs_keys if k in data}
kwargs = {k: data[k] for k in self.kwargs_keys}
if self.dask_load:
img = img.get_image_dask_data(**kwargs).compute()
else:
Expand Down

0 comments on commit 7ce9946

Please sign in to comment.