Skip to content
Snippets Groups Projects
Commit 1c45c415 authored by Yannick DAYER's avatar Yannick DAYER
Browse files

Only consider DelayedSample objects

parent e4c91296
No related branches found
No related tags found
1 merge request!65CSVDataset multiple samples per row
Pipeline #50733 passed
...@@ -89,7 +89,7 @@ class CSVToSampleLoader(TransformerMixin, BaseEstimator): ...@@ -89,7 +89,7 @@ class CSVToSampleLoader(TransformerMixin, BaseEstimator):
# samples can contain lists of samples, unpack if it's the case # samples can contain lists of samples, unpack if it's the case
unpacked_samples = [] unpacked_samples = []
for e in samples: for e in samples:
if isinstance(e, (Sample, DelayedSample)): if isinstance(e, DelayedSample):
unpacked_samples.append(e) unpacked_samples.append(e)
else: else:
unpacked_samples.extend(e) unpacked_samples.extend(e)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment