run black repo wide

sslivkoff · sslivkoff · commit ea2ebab80db3 · 2023-06-04T16:16:10.000-07:00
diff --git a/flood/__main__.py b/flood/__main__.py
@@ -5,4 +5,3 @@
 
 if __name__ == '__main__':
     cli_run.run_cli()
-
diff --git a/flood/cli/cli_run.py b/flood/cli/cli_run.py
@@ -48,4 +48,3 @@ def run_cli(raw_command: str | None = None) -> None:
         command_index=command_index,
         config=config,
     )
-
diff --git a/flood/cli/ls_command.py b/flood/cli/ls_command.py
@@ -36,4 +36,3 @@ def ls_command() -> None:
         text_style=flood.styles['metavar'],
     )
     toolstr.print_bullet(key='all', value='', colon_str='')
-
diff --git a/flood/cli/report_command.py b/flood/cli/report_command.py
@@ -41,7 +41,6 @@ def report_command(
     output: str | None,
     metrics: typing.Sequence[str] | None,
 ) -> None:
-
     if output is None:
         import os
 
@@ -52,4 +51,3 @@ def report_command(
         output_dir=output,
         metrics=metrics,
     )
-
diff --git a/flood/cli/root_command.py b/flood/cli/root_command.py
@@ -166,4 +166,3 @@ def root_command(
             output_dir=output_dir or True,
             figures=figures,
         )
-
diff --git a/flood/cli/samples_collect_command.py b/flood/cli/samples_collect_command.py
@@ -48,4 +48,3 @@ def samples_collect_command(
         sizes=sizes,
         datatypes=datatypes,
     )
-
diff --git a/flood/cli/samples_download_command.py b/flood/cli/samples_download_command.py
@@ -61,4 +61,3 @@ def download_samples_command(
         output_dir=output_dir,
         only_missing=missing,
     )
-
diff --git a/flood/cli/samples_ls_command.py b/flood/cli/samples_ls_command.py
@@ -57,4 +57,3 @@ def samples_ls_command() -> None:
                 print('-', filename)
             print()
             print('download using `flood samples download` command')
-
diff --git a/flood/generators/object_generators/address_generators.py b/flood/generators/object_generators/address_generators.py
@@ -29,4 +29,3 @@ def generate_eoas(
         n=n,
         random_seed=random_seed,
     )
-
diff --git a/flood/generators/object_generators/block_generators.py b/flood/generators/object_generators/block_generators.py
@@ -101,7 +101,9 @@ def _generate_block_ranges_strides(
         random_phases = list(range(range_size))
         rng.shuffle(random_phases)
         for random_phase in random_phases:
-            strides = np.arange(start_block + random_phase, end_block, range_size)
+            strides = np.arange(
+                start_block + random_phase, end_block, range_size
+            )
             rng.shuffle(strides)
             n_missing = n - len(block_ranges)
             if n_missing > 0:
@@ -180,4 +182,3 @@ def _generate_block_ranges_individual(
         ranges_list = list(ranges)
         rng.shuffle(ranges_list)
         return ranges_list
-
diff --git a/flood/generators/object_generators/call_generators.py b/flood/generators/object_generators/call_generators.py
@@ -594,4 +594,3 @@ def generate_calls_trace_replay_transaction_vm_trace(
         )
         for transaction_hash in transaction_hashes
     ]
-
diff --git a/flood/generators/object_generators/slot_generators.py b/flood/generators/object_generators/slot_generators.py
@@ -15,4 +15,3 @@ def generate_slots(
         n=n,
         random_seed=random_seed,
     )
-
diff --git a/flood/generators/object_generators/timing_generators.py b/flood/generators/object_generators/timing_generators.py
@@ -143,4 +143,3 @@ def _generate_timings_for_soak_test(
             raise Exception('must specify 1 duration for soak test')
 
     return rates, durations
-
diff --git a/flood/generators/object_generators/transaction_generators.py b/flood/generators/object_generators/transaction_generators.py
@@ -16,4 +16,3 @@ def generate_transaction_hashes(
         n=n,
         random_seed=random_seed,
     )
-
diff --git a/flood/generators/raw_data_sources/raw_data_spec.py b/flood/generators/raw_data_sources/raw_data_spec.py
@@ -6,7 +6,9 @@
 raw_data_url_template = (
     'https://datasets.paradigm.xyz/evm_samples/{network}_samples__{version}/'
 )
-raw_data_file_template = '{network}_{datatype}_samples__{size}__{version}.parquet'
+raw_data_file_template = (
+    '{network}_{datatype}_samples__{size}__{version}.parquet'
+)
 
 default_sizes = {
     'XS': 1_000,
@@ -22,4 +24,3 @@
     'transactions',
     'slots',
 ]
-
diff --git a/flood/generators/raw_data_sources/raw_download_utils.py b/flood/generators/raw_data_sources/raw_download_utils.py
@@ -69,4 +69,3 @@ def get_flood_samples_dir() -> str:
         return os.path.join(os.getcwd(), 'flood_samples')
     else:
         return flood_samples_dir
-
diff --git a/flood/generators/raw_data_sources/raw_gather_utils.py b/flood/generators/raw_data_sources/raw_gather_utils.py
@@ -199,4 +199,3 @@ def _write_raw_samples(
 
             # write file
             sized_data.write_parquet(output_path)
-
diff --git a/flood/generators/raw_data_sources/raw_sample_loading.py b/flood/generators/raw_data_sources/raw_sample_loading.py
@@ -166,4 +166,3 @@ def load_samples(
         return df[columns[0]].to_list()
     else:
         return df.rows()
-
diff --git a/flood/generators/rng_utils.py b/flood/generators/rng_utils.py
@@ -19,4 +19,3 @@ def get_rng(random_seed: spec.RandomSeed | None = None) -> np.random.Generator:
         return gen
     else:
         raise Exception('invalid seed format: ' + str(type(random_seed)))
-
diff --git a/flood/generators/test_generators/address_test_generators.py b/flood/generators/test_generators/address_test_generators.py
@@ -54,4 +54,3 @@ def generate_test_eth_get_transaction_count(
         duration=duration,
         durations=durations,
     )
-
diff --git a/flood/generators/test_generators/block_test_generators.py b/flood/generators/test_generators/block_test_generators.py
@@ -54,4 +54,3 @@ def generate_test_eth_get_block_by_number(
 #         duration=duration,
 #         durations=durations,
 #     )
-
diff --git a/flood/generators/test_generators/contract_test_generators.py b/flood/generators/test_generators/contract_test_generators.py
@@ -78,4 +78,3 @@ def generate_test_eth_call(
         duration=duration,
         durations=durations,
     )
-
diff --git a/flood/generators/test_generators/generic_test_generators.py b/flood/generators/test_generators/generic_test_generators.py
@@ -135,4 +135,3 @@ def generate_test(
 #         tests.append(test)
 
 #     return tests
-
diff --git a/flood/generators/test_generators/log_test_generators.py b/flood/generators/test_generators/log_test_generators.py
@@ -129,4 +129,3 @@ def generate_test_eth_get_logs(
 #         }
 
 #     return tests
-
diff --git a/flood/generators/test_generators/trace_test_generators.py b/flood/generators/test_generators/trace_test_generators.py
@@ -204,4 +204,3 @@ def generate_test_trace_replay_transaction_vm_trace(
         duration=duration,
         durations=durations,
     )
-
diff --git a/flood/generators/test_generators/transaction_test_generators.py b/flood/generators/test_generators/transaction_test_generators.py
@@ -53,4 +53,3 @@ def generate_test_eth_get_transaction_receipt(
         duration=duration,
         durations=durations,
     )
-
diff --git a/flood/spec.py b/flood/spec.py
@@ -180,4 +180,3 @@ class SingleRunResultsPayload(typing.TypedDict):
         test: LoadTest
         nodes: Nodes
         results: typing.Mapping[str, LoadTestOutput]
-
diff --git a/flood/tests/__init__.py b/flood/tests/__init__.py
@@ -1,4 +1,3 @@
 from .equality_tests import *
 from .load_tests import *
 from .runner import *
-
diff --git a/flood/tests/equality_tests/__init__.py b/flood/tests/equality_tests/__init__.py
@@ -1,2 +1 @@
-
 from .equality_test_runs import *
diff --git a/flood/tests/equality_tests/equality_test_runs.py b/flood/tests/equality_tests/equality_test_runs.py
@@ -127,7 +127,9 @@ def run_equality_test(
         for name in sorted(failed):
             flood.print_bullet(key=name, value='', colon_str='')
     print()
-    toolstr.print('summary saved to: ' + file_path, style=flood.styles['comment'])
+    toolstr.print(
+        'summary saved to: ' + file_path, style=flood.styles['comment']
+    )
 
 
 def _json_equal(lhs: typing.Any, rhs: typing.Any) -> bool:
@@ -251,7 +253,6 @@ def _print_result_diff(
             n_printed = 0
             for i, (item0, item1) in enumerate(zip(result0, result1)):
                 if not _json_equal(item0, item1):
-
                     if n_printed >= 10:
                         print('...')
                         break
@@ -273,4 +274,3 @@ def _print_result_diff(
             flood.print_header('differences in values')
             for node, result in zip(nodes, results):
                 flood.print_bullet(key=node['name'], value=result)
-
diff --git a/flood/tests/equality_tests/equality_test_sets.py b/flood/tests/equality_tests/equality_test_sets.py
@@ -278,4 +278,3 @@ def get_trace_equality_tests(
             {},
         ),
     ]
-
diff --git a/flood/tests/load_tests/load_test_construction.py b/flood/tests/load_tests/load_test_construction.py
@@ -86,4 +86,3 @@ def create_load_test(
         load_test.append(attack)
 
     return load_test
-
diff --git a/flood/tests/load_tests/load_test_plots.py b/flood/tests/load_tests/load_test_plots.py
@@ -201,4 +201,3 @@ def plot_load_test_result_metrics(
         ylabel=ylabel,
     )
     plt.legend(loc='center right')
-
diff --git a/flood/tests/load_tests/load_test_reports.py b/flood/tests/load_tests/load_test_reports.py
@@ -372,4 +372,3 @@ def _create_test_chunk(
         'inputs': [],
     },
 ]
-
diff --git a/flood/tests/load_tests/load_test_runs.py b/flood/tests/load_tests/load_test_runs.py
@@ -48,12 +48,16 @@ def run_load_tests(
     # case: single node and multiple tests
     elif node is not None and tests is not None:
         for name, test in tqdm.tqdm(tests.items(), **pbar):
-            results[name] = schedule_load_test(node=node, verbose=verbose, test=test)
+            results[name] = schedule_load_test(
+                node=node, verbose=verbose, test=test
+            )
 
     # case: multiple nodes and single tests
     elif nodes is not None and test is not None:
         for name, nd in tqdm.tqdm(nodes.items(), **pbar):
-            results[name] = schedule_load_test(node=nd, verbose=verbose, test=test)
+            results[name] = schedule_load_test(
+                node=nd, verbose=verbose, test=test
+            )
 
     # case: multiple nodes and multiple tests
     elif nodes is not None and tests is not None:
@@ -337,4 +341,3 @@ def _run_load_test_remotely(
     with open(results_path, 'r') as f:
         results: spec.SingleRunResultsPayload = json.load(f)
         return results['results'][node['name']]
-
diff --git a/flood/tests/load_tests/vegeta.py b/flood/tests/load_tests/vegeta.py
@@ -167,4 +167,3 @@ def _create_vegeta_report(
         'last_response_timestamp': report['end'],
         'final_wait_time': report['wait'] / 1e9,
     }
-
diff --git a/flood/tests/runner.py b/flood/tests/runner.py
@@ -434,4 +434,3 @@ def _print_single_run_conclusion_copy(
         metrics=metrics,
         indent=4,
     )
-
diff --git a/flood/user_io/file_io.py b/flood/user_io/file_io.py
@@ -125,4 +125,3 @@ def load_single_run_results(
 ) -> typing.Mapping[str, flood.LoadTestOutput]:
     payload = load_single_run_results_payload(output_dir=output_dir)
     return payload['results']
-
diff --git a/flood/user_io/inputs.py b/flood/user_io/inputs.py
@@ -210,4 +210,3 @@ def parse_nodes_network(nodes: typing.Mapping[str, spec.Node]) -> str:
         return network
     else:
         raise Exception('need str network')
-
diff --git a/flood/user_io/notebook_io.py b/flood/user_io/notebook_io.py
@@ -25,7 +25,9 @@ class NotebookCellTemplateFunction(typing.TypedDict):
 
     class NotebookChunkTemplate(typing.TypedDict):
         type: typing.Literal['chunk']
-        f: typing.Callable[..., typing.Sequence[nbformat.notebooknode.NotebookNode]]
+        f: typing.Callable[
+            ..., typing.Sequence[nbformat.notebooknode.NotebookNode]
+        ]
         inputs: typing.Sequence[str]
 
     NotebookCellPrecursor = typing.Union[
@@ -133,4 +135,3 @@ def create_notebook(
     # write notebook to file
     with open(output_path, 'w') as f:
         nbformat.write(notebook, f)  # type: ignore
-
diff --git a/flood/user_io/outputs.py b/flood/user_io/outputs.py
@@ -61,7 +61,6 @@ def get_nodes_plot_colors(
     colors = {}
     taken = set()
     for node in nodes.values():
-
         # print version
         version = node['client_version']
         if version is None:
@@ -244,4 +243,3 @@ def print_multiline_table(*args: typing.Any, **kwargs: typing.Any) -> None:
         label_style=flood.styles.get('metavar'),
         border=flood.styles.get('content'),
     )
-

Original file line number	Diff line number	Diff line change
`@@ -5,4 +5,3 @@`
`5`	`5`
`6`	`6`	`if __name__ == '__main__':`
`7`	`7`	`cli_run.run_cli()`
`8`		`-`
Original file line number	Diff line number	Diff line change
`@@ -48,4 +48,3 @@ def run_cli(raw_command: str \| None = None) -> None:`
`48`	`48`	`command_index=command_index,`
`49`	`49`	`config=config,`
`50`	`50`	`)`
`51`		`-`
Original file line number	Diff line number	Diff line change
`@@ -36,4 +36,3 @@ def ls_command() -> None:`
`36`	`36`	`text_style=flood.styles['metavar'],`
`37`	`37`	`)`
`38`	`38`	`toolstr.print_bullet(key='all', value='', colon_str='')`
`39`		`-`
Original file line number	Diff line number	Diff line change
`@@ -166,4 +166,3 @@ def root_command(`
`166`	`166`	`output_dir=output_dir or True,`
`167`	`167`	`figures=figures,`
`168`	`168`	`)`
`169`		`-`
Original file line number	Diff line number	Diff line change
`@@ -48,4 +48,3 @@ def samples_collect_command(`
`48`	`48`	`sizes=sizes,`
`49`	`49`	`datatypes=datatypes,`
`50`	`50`	`)`
`51`		`-`
Original file line number	Diff line number	Diff line change
`@@ -61,4 +61,3 @@ def download_samples_command(`
`61`	`61`	`output_dir=output_dir,`
`62`	`62`	`only_missing=missing,`
`63`	`63`	`)`
`64`		`-`
Original file line number	Diff line number	Diff line change
`@@ -29,4 +29,3 @@ def generate_eoas(`
`29`	`29`	`n=n,`
`30`	`30`	`random_seed=random_seed,`
`31`	`31`	`)`
`32`		`-`