ensure ndarray is writeable when -> tensor (#137)

Signed-off-by: nstarman <[email protected]>
GalOrrery · Sep 7, 2023 · d49368e · d49368e
1 parent 360593e
commit d49368e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/stream_ml/pytorch/_connect/data.py b/src/stream_ml/pytorch/_connect/data.py
@@ -51,7 +51,9 @@ def _from_ndarraytype_to_tensor(
         data: Data[np.ndarray[Any, Any]], /, **kwargs: Any
     ) -> Data[xp.Tensor]:
         """Convert from numpy.ndarray to torch.Tensor."""
-        return replace(data, array=xp.asarray(np.asarray(data.array), **kwargs))
+        array = np.array(data.array, copy=True, subok=False)
+        array.flags.writeable = True
+        return replace(data, array=xp.asarray(array, **kwargs))
 
     ASTYPE_REGISTRY[
         (asdf.tags.core.ndarray.NDArrayType, xp.Tensor)