Clean-up

OP2 · Jul 26, 2022 · cb3ae29 · cb3ae29
1 parent e09cd25
commit cb3ae29
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/pyop2/types/dat.py b/pyop2/types/dat.py
@@ -1015,13 +1015,12 @@ def vec_context(self, access):
         # values
         if access is not Access.WRITE:
             offset = 0
-            array = self._vec.array
-            for d in self:
-                with d.vec_ro as v:
-                    size = v.local_size
-                    array[offset:offset+size] = v.array_r[:]
-                    offset += size
-            del array
+            with self._vec as array:
+                for d in self:
+                    with d.vec_ro as v:
+                        size = v.local_size
+                        array[offset:offset+size] = v.array_r[:]
+                        offset += size
 
         yield self._vec
         if access is not Access.READ: