add illiffe astype function

AWehrhahn · Apr 6, 2022 · 441f3f8 · 441f3f8
1 parent db6d338
commit 441f3f8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/pysme/iliffe_vector.py b/src/pysme/iliffe_vector.py
@@ -273,6 +273,11 @@ def from_indices(cls, array, indices):
             arr = [array[l:u] for l, u in zip(offsets[:-1], offsets[1:])]
             return cls(arr)
 
+    def astype(self, dtype):
+        data = self.data.astype(dtype)
+        offsets = self.offsets
+        return Iliffe_vector(data, offsets, dtype=dtype)
+
     # For IO with Flex
     def _prepare(self, name: str):
         cls = self.__class__