Skip to content
Snippets Groups Projects
Commit 3dcbd8b6 authored by Alán Muñoz's avatar Alán Muñoz
Browse files

bugfix

Former-commit-id: d2ea2a90b4f84d7ee719fec316322ed726bc8da3
parent 20d8a81f
No related branches found
No related tags found
No related merge requests found
from pydoc import locate import h5py
from typing import List, Dict, Union from typing import List, Dict, Union
from pydoc import locate
import numpy as np
import pandas as pd import pandas as pd
from postprocessor.core.processes.base import ParametersABC from postprocessor.core.processes.base import ParametersABC
...@@ -85,7 +88,6 @@ class PostProcessor: ...@@ -85,7 +88,6 @@ class PostProcessor:
""" """
Dynamically import a process class from the 'processes' folder. Dynamically import a process class from the 'processes' folder.
Assumes process filename and class name are the same Assumes process filename and class name are the same
# TODO add support for passing parameters
""" """
return locate("postprocessor.core.processes." + process + "." + process) return locate("postprocessor.core.processes." + process + "." + process)
...@@ -94,7 +96,6 @@ class PostProcessor: ...@@ -94,7 +96,6 @@ class PostProcessor:
""" """
Dynamically import a process class from the 'processes' folder. Dynamically import a process class from the 'processes' folder.
Assumes process filename and class name are the same Assumes process filename and class name are the same
# TODO add support for passing parameters
""" """
return locate( return locate(
"postprocessor.core.processes." + process + "." + process + "Parameters" "postprocessor.core.processes." + process + "." + process + "Parameters"
...@@ -103,7 +104,7 @@ class PostProcessor: ...@@ -103,7 +104,7 @@ class PostProcessor:
def run(self): def run(self):
new_ids = self.merger.run(self._signal[self.processes["merger"]]) new_ids = self.merger.run(self._signal[self.processes["merger"]])
for name, ids in new_ids.items(): for name, ids in new_ids.items():
self._writer.write(ids, "/postprocessing/cell_info/" + name) self._writer.write("/postprocessing/cell_info/" + name, ids)
picks = self.picker.run(self._signal[self.processes["picker"][0]]) picks = self.picker.run(self._signal[self.processes["picker"][0]])
for process, datasets in self.processes["processes"].items(): for process, datasets in self.processes["processes"].items():
parameters = ( parameters = (
...@@ -118,7 +119,6 @@ class PostProcessor: ...@@ -118,7 +119,6 @@ class PostProcessor:
if isinstance(dataset, list): # multisignal process if isinstance(dataset, list): # multisignal process
signal = [self._signal[d] for d in dataset] signal = [self._signal[d] for d in dataset]
elif isinstance(dataset, str): elif isinstance(dataset, str):
print(dataset)
signal = self._signal[dataset] signal = self._signal[dataset]
else: else:
raise ("Incorrect dataset") raise ("Incorrect dataset")
...@@ -145,7 +145,10 @@ class PostProcessor: ...@@ -145,7 +145,10 @@ class PostProcessor:
elif isinstance(dataset, str): elif isinstance(dataset, str):
outpath = dataset[1:].replace("/", "_") outpath = dataset[1:].replace("/", "_")
self._writer.write(result, "/postprocessing/" + process + "/" + outpath) def write_result(
self, result: Union[List, pd.DataFrame, np.ndarray], path: str, metadata: Dict
):
self._writer.write(result, "/postprocessing/" + process + "/" + outpath)
def _if_dict(item): def _if_dict(item):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment