File tree Expand file tree Collapse file tree 1 file changed +13
-2
lines changed
Expand file tree Collapse file tree 1 file changed +13
-2
lines changed Original file line number Diff line number Diff line change 9494
9595def 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 :
You can’t perform that action at this time.
0 commit comments