From 41b75885da926acc72dc39212d0b5d1ca4642719 Mon Sep 17 00:00:00 2001 From: Riyaz Haque Date: Sun, 6 Oct 2024 10:41:47 -0700 Subject: [PATCH] Fix lint formatiing --- .../experiments/amg2023/experiment.py | 20 ++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/var/exp_repo/experiments/amg2023/experiment.py b/var/exp_repo/experiments/amg2023/experiment.py index 2d80c48af..30e282017 100644 --- a/var/exp_repo/experiments/amg2023/experiment.py +++ b/var/exp_repo/experiments/amg2023/experiment.py @@ -99,18 +99,28 @@ def compute_applications_section(self): variables[ny] = initial_n variables[nz] = initial_n zips_size = "size" - experiment_setup["zips"] = {f"{zips_size}" : [nx, ny, nz]} + experiment_setup["zips"] = {f"{zips_size}": [nx, ny, nz]} if self.spec.satisfies("programming_model=openmp"): - experiment_setup["matrices"] = [{"size_nodes_threads": [f"{zips_size}", "n_nodes", "n_threads_per_proc"]}] + experiment_setup["matrices"] = [ + { + "size_nodes_threads": [ + f"{zips_size}", + "n_nodes", + "n_threads_per_proc", + ] + } + ] elif self.spec.satisfies("programming_model=cuda") or self.spec.satisfies( "programming_model=rocm" ): experiment_setup["matrix"] = [f"{zips_size}"] if self.spec.satisfies("programming_model=openmp"): - experiment_setup["exclude"] = {"where" : [ - "{n_threads_per_proc} * {n_ranks} > {n_nodes} * {sys_cores_per_node}" - ]} + experiment_setup["exclude"] = { + "where": [ + "{n_threads_per_proc} * {n_ranks} > {n_nodes} * {sys_cores_per_node}" + ] + } else: input_params = {} if self.spec.satisfies("experiment=throughput"):