Skip to content

Commit d7af1b0

Browse files
author
Austin Glover
committed
format
1 parent 35a1d36 commit d7af1b0

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

tests/benchmark.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,12 @@
9494

9595
def benchmark_uvu(params):
9696
def get_problems():
97-
return mace_problems() + nequip_problems() + nequix_problems() + seven_net_problems()
97+
return (
98+
mace_problems()
99+
+ nequip_problems()
100+
+ nequix_problems()
101+
+ seven_net_problems()
102+
)
98103

99104
float64_problems = get_problems()
100105
for problem in float64_problems:
@@ -285,8 +290,14 @@ def benchmark_convolution(params):
285290
graphs = download_graphs(params, filenames)
286291

287292
if not params.disable_bench:
293+
288294
def get_problems():
289-
return mace_problems() + nequip_problems() + nequix_problems() + seven_net_problems()
295+
return (
296+
mace_problems()
297+
+ nequip_problems()
298+
+ nequix_problems()
299+
+ seven_net_problems()
300+
)
290301

291302
float64_problems = get_problems()
292303
for problem in float64_problems:

0 commit comments

Comments
 (0)