From 385399c6f3e51d8d665fc59416e2a2ee87dc076c Mon Sep 17 00:00:00 2001
From: BO ZHANG <bozhang@nao.cas.cn>
Date: Wed, 20 Dec 2023 14:12:48 +0800
Subject: [PATCH] make alias: Pipeline.retry

---
 csst_common/pipeline.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/csst_common/pipeline.py b/csst_common/pipeline.py
index e93f4e0..3086189 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."""
-- 
GitLab