diff --git a/extraction/core/extractor.py b/extraction/core/extractor.py index e36bf2bbc46345f85635999e663fddccd5817849..39713f2ae9056ac1651471dcf64a06e953a7514a 100644 --- a/extraction/core/extractor.py +++ b/extraction/core/extractor.py @@ -430,6 +430,12 @@ class Extractor(ProcessABC): elif channel in self.img_bgsub: return self.img_bgsub[channel] + def run_tp(self, tp, **kwargs): + """ + Wrapper to add compatiblibility with other pipeline steps + """ + return self.run(tps=[tp], **kwargs) + def run(self, tree=None, tps: List[int] = None, save=True, **kwargs) -> dict: if tree is None: