diff --git a/core/processes/base.py b/core/processes/base.py deleted file mode 100644 index 9dadb0470eab4de33f56538e9008382496415fda..0000000000000000000000000000000000000000 --- a/core/processes/base.py +++ /dev/null @@ -1,54 +0,0 @@ -from pathlib import Path, PosixPath -from typing import Union -from abc import ABC, abstractmethod - -from yaml import safe_load, dump - - -class ParametersABC(ABC): - """ - Base class to add yaml functionality to parameters - - """ - - def to_dict(self): - return self.__dict__ - - @classmethod - def from_dict(cls, d: dict): - return cls(**d) - - def to_yaml(self, path: Union[PosixPath, str] = None): - if path: - with open(Path(path), "w") as f: - dump(self.to_dict(), f) - return dump(self.to_dict()) - - @classmethod - def from_yaml(cls, path: Union[PosixPath, str]): - with open(Path(file)) as f: - params = safe_load(f) - return cls(**params) - - @classmethod - @abstractmethod - def default(cls): - pass - - -class ProcessABC(ABC): - "Base class for processes" - - def __init__(self, parameters): - self._parameters = parameters - - for k, v in parameters.to_dict().items(): # access parameters directly - setattr(self, k, v) - - @property - def parameters(self): - return self._parameters - - @abstractmethod - def run(self): - pass diff --git a/core/processes/merger.py b/core/processes/merger.py index 51a91bf98e22e2c9e4c82bac10830494a839b513..b0f411b7f9baea94aadcbd47f62ea4171a179b7d 100644 --- a/core/processes/merger.py +++ b/core/processes/merger.py @@ -1,4 +1,4 @@ -from postprocessor.core.processes.base import ParametersABC, ProcessABC +from agora.base import ParametersABC, ProcessABC from postprocessor.core.functions.tracks import clean_tracks, merge_tracks, join_tracks diff --git a/core/processes/picker.py b/core/processes/picker.py index 9ddd72813147f10dca026872af5871493324aefb..6b0ae14065c311a7c497e86ada74a5bb9da77d1f 100644 --- a/core/processes/picker.py +++ b/core/processes/picker.py @@ -6,7 +6,7 @@ import pandas as pd from core.cells import CellsHDF -from postprocessor.core.processes.base import ParametersABC, ProcessABC +from agora.base import ParametersABC, ProcessABC from postprocessor.core.functions.tracks import max_ntps, max_nonstop_ntps diff --git a/core/processor.py b/core/processor.py index 7421a9bb802773e8f73f292974076aea3e08ddcc..d51a29108eefa45325b9371e1664a0c2e73b6f3e 100644 --- a/core/processor.py +++ b/core/processor.py @@ -1,7 +1,7 @@ from typing import List, Dict, Union import pandas as pd -from postprocessor.core.processes.base import ParametersABC +from agora.base import ParametersABC from postprocessor.core.processes.merger import MergerParameters, Merger from postprocessor.core.processes.picker import PickerParameters, Picker from postprocessor.core.io.writer import Writer