We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f352a71 commit 7b284e9Copy full SHA for 7b284e9
sup3r/preprocessing/samplers/base.py
@@ -14,7 +14,7 @@
14
uniform_box_sampler,
15
uniform_time_sampler,
16
)
17
-from sup3r.preprocessing.utilities import lowered
+from sup3r.preprocessing.utilities import compute_if_dask, lowered
18
19
logger = logging.getLogger(__name__)
20
@@ -249,9 +249,7 @@ def _compute_samples(self, samples):
249
"""
250
if self.mode == 'eager':
251
return samples
252
- if isinstance(samples, tuple):
253
- return tuple(np.asarray(s) for s in samples)
254
- return np.asarray(samples)
+ return compute_if_dask(samples)
255
256
def _fast_batch(self):
257
"""Get batch of samples with adjacent time slices."""
0 commit comments