Skip to content
Snippets Groups Projects
Commit 36ccaf35 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Cleaning up

parent fd9a4d9e
No related branches found
No related tags found
1 merge request!60CSVSampleLoaders as transformers
Pipeline #46545 passed
...@@ -94,14 +94,6 @@ class CSVToSampleLoader(TransformerMixin, BaseEstimator): ...@@ -94,14 +94,6 @@ class CSVToSampleLoader(TransformerMixin, BaseEstimator):
if not "path" in header: if not "path" in header:
raise ValueError("The field `path` is not available in your dataset.") raise ValueError("The field `path` is not available in your dataset.")
def __call__(self, f):
f.seek(0)
reader = csv.reader(f)
header = next(reader)
self.check_header(header)
return [self.convert_row_to_sample(row, header) for row in reader]
def convert_row_to_sample(self, row, header): def convert_row_to_sample(self, row, header):
path = row[0] path = row[0]
reference_id = row[1] reference_id = row[1]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment