diff --git a/csst_common/pipeline.py b/csst_common/pipeline.py index e93f4e04e15849305bbe088bdc64aa44b5ca5590..3086189fb1bcc38939f8cead4034008a0a1717bf 100644 --- a/csst_common/pipeline.py +++ b/csst_common/pipeline.py @@ -17,9 +17,9 @@ import subprocess import traceback import warnings from typing import Callable, NamedTuple, Optional, Any - from astropy.time import Time, TimeDelta +from .utils import retry from .ccds import CCDS from .dfs import DFS from .file import File @@ -27,7 +27,6 @@ from .logger import get_logger from .status import CsstStatus, CsstResult -# TODO: unit test in need class Pipeline: """ CSST pipeline configuration class. @@ -223,6 +222,10 @@ class Pipeline: output=output, ) + @staticmethod + def retry(*args, **kwargs): + return retry(*args, **kwargs) + # class ErrorTrace: # """Write error trace to file."""