* after pre-commit

Author: HYLcool

trinity/cli/client.py

@@ -39,4 +39,4 @@ def request(url, **kwargs):
     )
     if res:
         print(res)
-        print(res['message'])
+        print(res["message"])

trinity/cli/launcher.py

@@ -111,6 +111,7 @@ def activate_data_module(config_path: str):
         logger.error(f"Failed to activate data module: {res['return_msg']}.")
         return
 
+
 def run(config_path: str):
     # TODO: support parse all args from command line
     config = load_config(config_path)

trinity/cli/server.py

@@ -1,11 +1,13 @@
 import traceback
+
 from flask import Flask, jsonify, request
 
 app = Flask(__name__)
 
-APP_NAME = 'trinity_training'
+APP_NAME = "trinity_training"
 PORT = 5006
 
+
 @app.route(f"/{APP_NAME}", methods=["GET"])
 def trinity_training():
     config_path = request.args.get("configPath")
@@ -15,7 +17,7 @@ def trinity_training():
         run(config_path)
         ret = 0
         msg = "Training Success."
-    except:
+    except:  # noqa: E722
         traceback.print_exc()
         msg = traceback.format_exc()
         ret = 1

trinity/data/core/dataset_db.py

@@ -5,10 +5,10 @@
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy.pool import NullPool
 
+from trinity.buffer.utils import retry_session
 from trinity.common.config import DataConfig
 from trinity.common.schema import Base, RftDatasetModel
 from trinity.data.core.dataset import RftDataset
-from trinity.buffer.utils import retry_session
 from trinity.utils.log import get_logger
 
 logger = get_logger(__name__)
@@ -55,7 +55,9 @@ def __init__(self, config: DataConfig) -> None:
         self.session = sessionmaker(bind=self.engine)
 
     def add_entries(self, dataset: RftDataset):
-        with retry_session(self) as session:
+        with retry_session(
+            self, self.config.max_retry_times, self.config.max_retry_interval
+        ) as session:
             session.add_all(rft_dataset_to_model(dataset))
 
     def get_entries(self, num_entries: int, order_by: str = None, ascending: bool = False):
@@ -65,7 +67,9 @@ def get_entries(self, num_entries: int, order_by: str = None, ascending: bool =
             order_by_key = asc(order_by_key) if ascending else desc(order_by_key)
         else:
             order_by_key = None
-        with retry_session(self) as session:
+        with retry_session(
+            self, self.config.max_retry_times, self.config.max_retry_interval
+        ) as session:
             entries = (
                 session.query(RftDatasetModel)
                 .order_by(order_by_key)

trinity/data/server.py

@@ -2,9 +2,10 @@
 
 app = Flask(__name__)
 
-APP_NAME = 'data_workflow'
+APP_NAME = "data_workflow"
 PORT = 5005
 
+
 @app.route(f"/{APP_NAME}", methods=["GET"])
 def data_workflow():
     from trinity.common.config import load_config