FIX asarray

gallantlab · Feb 16, 2023 · 2d3dc49 · 2d3dc49
1 parent a30dbe8
commit 2d3dc49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/himalaya/backend/torch.py b/himalaya/backend/torch.py
@@ -176,7 +176,7 @@ def asarray(x, dtype=None, device="cpu"):
         tensor = torch.as_tensor(x, dtype=dtype, device=device)
     except Exception:
         import numpy as np
-        if x.device != "cpu":
+        if torch.is_tensor(x) and x.device != "cpu":
             x = x.cpu()
         array = np.asarray(x, dtype=_dtype_to_str(dtype))
         tensor = torch.as_tensor(array, dtype=dtype, device=device)