linting and formatting

Author: erwin-wee

chainbench/main.py

@@ -210,7 +210,11 @@ def validate_profile_path(ctx: Context, param: Parameter, value: str) -> str:
 @click.option("--pg-password", default=None, help="PG password")
 @click.option("--use-latest-blocks", is_flag=True, help="Uses latest blocks for test data")
 @click.option("--size", default=None, help="Set the size of the test data. e.g. --size S")
-@click.option("--ref-url", default=None, help="Reference Node URL for retrieving test data before test starts. If not specified, target url is used instead.")
+@click.option(
+    "--ref-url",
+    default=None,
+    help="Reference Node URL for retrieving test data before test starts. If not specified, target url is used.",
+)
 @click.pass_context
 def start(
     ctx: Context,
@@ -282,7 +286,7 @@ def start(
 
     user_classes = {}
     for locustfile in parse_locustfile_paths([final_profile_path.__str__()]):
-        _user_classes, _  = load_locustfile(locustfile)
+        _user_classes, _ = load_locustfile(locustfile)
         for key, value in _user_classes.items():
             user_classes[key] = value
     test_data_types = set()

chainbench/util/event.py

@@ -55,7 +55,7 @@ def cli_custom_arguments(parser: LocustArgumentParser):
         "--ref-url",
         type=str,
         default=None,
-        help="Set the reference node url to be used for retrieving test data before test starts. If empty, defaults to target url.",
+        help="Reference node url used for fetching test data before test starts. If empty, defaults to target url.",
         include_in_web_ui=True,
     )
 
@@ -195,7 +195,11 @@ def on_init(environment: Environment, **_kwargs):
                     logger.info(f"Initializing test data for {test_data_class_name}")
                     print(f"Initializing test data for {test_data_class_name}")
                     if environment.parsed_options:
-                        ref_node_url = environment.parsed_options.ref_url if environment.parsed_options.ref_url is not None else environment.host
+                        ref_node_url = (
+                            environment.parsed_options.ref_url
+                            if environment.parsed_options.ref_url is not None
+                            else environment.host
+                        )
                         user_test_data.init_http_client(ref_node_url)
                         if isinstance(user_test_data, EvmTestData):
                             chain_id: ChainId = user_test_data.fetch_chain_id()