MFE: fixed return output of the NumPy version

bh107 · Apr 7, 2017 · 1f679be · 1f679be
1 parent 7504938
commit 1f679be
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/benchmarks/magnetic_field_extrapolation/python_numpy/magnetic_field_extrapolation.py b/benchmarks/magnetic_field_extrapolation/python_numpy/magnetic_field_extrapolation.py
@@ -66,7 +66,8 @@ def main():
 
     B.start()
     for _ in range(B.size[2]):
-        calcB(window(B_x0))
+        Rx, Ry, Rz = calcB(window(B_x0))
+        B.flush()
     B.stop()
     B.pprint()