|
@@ -107,7 +107,7 @@ def crop_patches(crop_size,
|
|
if max_workers < 0:
|
|
if max_workers < 0:
|
|
raise ValueError("`max_workers` must be a non-negative integer!")
|
|
raise ValueError("`max_workers` must be a non-negative integer!")
|
|
|
|
|
|
- if subset is None:
|
|
|
|
|
|
+ if subsets is None:
|
|
subsets = ('', )
|
|
subsets = ('', )
|
|
|
|
|
|
if max_workers == 0:
|
|
if max_workers == 0:
|
|
@@ -280,4 +280,4 @@ def random_split(samples,
|
|
# Append remainder to the last split
|
|
# Append remainder to the last split
|
|
splits[-1].append(splits[ed_idx:])
|
|
splits[-1].append(splits[ed_idx:])
|
|
|
|
|
|
- return splits
|
|
|
|
|
|
+ return splits
|