Add to .cpu()

ecmwf · Jan 9, 2025 · 75ea925 · 75ea925
1 parent d040474
commit 75ea925
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/earthkit/data/utils/array.py b/src/earthkit/data/utils/array.py
@@ -148,7 +148,7 @@ def module(self):
         return torch
 
     def to_numpy(self, v):
-        return v.numpy()
+        return v.cpu().numpy()
 
     def from_numpy(self, v):
         import torch