Skip to content
Snippets Groups Projects
Commit c0a1abee authored by François  LAURENT's avatar François LAURENT
Browse files

BackendExplorer.prepare_labels handles multiple directories...

BackendExplorer.prepare_labels handles multiple directories (larvatagger.jl#72)
parent d6a46221
No related branches found
No related tags found
No related merge requests found
Pipeline #92848 failed
...@@ -372,14 +372,14 @@ run `poetry add {pkg}` from directory: \n ...@@ -372,14 +372,14 @@ run `poetry add {pkg}` from directory: \n
source.unlink() source.unlink()
check_permissions(destination) check_permissions(destination)
def list_input_files(self, query=None): def list_input_files(self, **kwargs):
return self.list_files(self.raw_data_dir(), query) return self.list_files(self.raw_data_dir(), **kwargs)
def list_interim_files(self, query=None): def list_interim_files(self, **kwargs):
return self.list_files(self.interim_data_dir(), query) return self.list_files(self.interim_data_dir(), **kwargs)
def list_output_files(self, query=None): def list_output_files(self, **kwargs):
return self.list_files(self.processed_data_dir(), query) return self.list_files(self.processed_data_dir(), **kwargs)
def list_model_files(self, query=None): def list_model_files(self, query=None):
return self.list_files(self.model_dir(), query) return self.list_files(self.model_dir(), query)
...@@ -399,11 +399,17 @@ run `poetry add {pkg}` from directory: \n ...@@ -399,11 +399,17 @@ run `poetry add {pkg}` from directory: \n
if group_by_directories: if group_by_directories:
files_ = defaultdict(list) files_ = defaultdict(list)
for file in files: for file in files:
files_[file.parent.name].append(file) files_[str(file.parent)].append(file)
files = dict(files_) files = dict(files_)
return files return files
def prepare_labels(self, input_files): def prepare_labels(self, input_files):
if isinstance(input_files, dict):
input_files_and_labels = dict()
for parent in input_files.keys():
input_files_and_labels[parent] = self.prepare_labels(input_files[parent])
return input_files_and_labels
#
from .data.labels import Labels, labels_file_extension from .data.labels import Labels, labels_file_extension
labels = Labels() labels = Labels()
# load and remove any *metadata* file. # load and remove any *metadata* file.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment