Skip to content

Commit

Permalink
Linting
Browse files Browse the repository at this point in the history
  • Loading branch information
Alexsandruss committed Dec 13, 2024
1 parent fd2662f commit 4412505
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions sklbench/runner/implementation.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,8 +106,12 @@ def run_benchmarks(args: argparse.Namespace) -> int:
with Pool(n_proc) as pool:
datasets = pool.map(load_data, dataset_cases.values())
if args.describe_datasets:
for ((data, data_description), data_name) in zip(datasets, dataset_cases.keys()):
print(f"{data_name}:\n\tshape: {data['x'].shape}\n\tparameters: {data_description}")
for (data, data_description), data_name in zip(
datasets, dataset_cases.keys()
):
print(
f"{data_name}:\n\tshape: {data['x'].shape}\n\tparameters: {data_description}"
)
sys.exit(0)
# free memory used by prefetched datasets
del datasets
Expand Down

0 comments on commit 4412505

Please sign in to comment.