Merge branch 'master' into generalised-zhang-li

ubermag · Jul 14, 2024 · 5d59e57 · 5d59e57
2 parents 52bf30a + 6f69f1c
commit 5d59e57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/oommfc/compute.py b/oommfc/compute.py
@@ -160,11 +160,11 @@ def compute(
             col = [
                 c for c in table.data.columns if c.endswith(":evolver:Total energy")
             ][0]
-            output = table.data[col][0]
+            output = table.data[col][0].item()
         else:
             output = table.data[
                 f"{oxs_class(func.__self__, system)}:{func.__self__.name}:Energy"
-            ][0]
+            ][0].item()
     else:
         output = df.Field.from_file(output_file)
         with contextlib.suppress(FileNotFoundError):