update retry

csst_common/utils/_retry.py

@@ -17,6 +17,7 @@ def retry(func, *args, **kwargs):
     else:
         n_try = 5
     # is DFS?
+    # TODO: support DFS2
     is_dfs = func.__class__.__module__.startswith("csst_dfs_client")
     # start trials
     for attempt in range(n_try):
@@ -24,7 +25,8 @@ def retry(func, *args, **kwargs):
         try:
             res = func(*args, **kwargs)
             if is_dfs:
-                assert res["code"] == 0, res
+                # TODO: support DFS2
+                assert res.success, res
             return res
         except Exception as e:
             # print error messages