From e69455ad353678479ce9b901823d2aad6d958772 Mon Sep 17 00:00:00 2001
From: "James R. Maddison" <j.r.maddison@ed.ac.uk>
Date: Tue, 28 May 2024 10:45:30 +0100
Subject: [PATCH] Fix

---
 tlm_adjoint/block_system.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/tlm_adjoint/block_system.py b/tlm_adjoint/block_system.py
index a6164cb1..66a67b9d 100644
--- a/tlm_adjoint/block_system.py
+++ b/tlm_adjoint/block_system.py
@@ -70,8 +70,8 @@
 """
 
 from .interface import (
-    comm_dup_cached, packed, space_comm, space_default_space_type, space_eq,
-    space_new, var_assign, var_locked, var_zero)
+    Packed, comm_dup_cached, packed, space_comm, space_default_space_type,
+    space_eq, space_new, var_assign, var_locked, var_zero)
 from .manager import manager_disabled
 from .petsc import (
     PETScOptions, PETScVec, PETScVecInterface, attach_destroy_finalizer)
@@ -113,7 +113,7 @@ def expand(e):
 
     q = deque(map(expand, iterable))
     while len(q) > 0:
-        e = packed(q.popleft())
+        e = Packed(q.popleft())
         if e.is_packed:
             e, = e
             yield e
@@ -125,7 +125,7 @@ def tuple_sub(iterable, sequence):
     iterator = iter_sub(iterable)
 
     def tuple_sub(iterator, value):
-        value = packed(value)
+        value = Packed(value)
         if value.is_packed:
             return next(iterator)
         else:
@@ -859,18 +859,20 @@ def solve(self, u, b, *,
             the solution.
         """
 
-        u = packed(u)
-        b = packed(b)
+        u_packed = Packed(u)
+        b_packed = Packed(b)
+        u = tuple(u_packed)
+        b = tuple(b_packed)
 
         pc_fn = self._pc_fn
-        if u.is_packed:
+        if u_packed.is_packed:
             pc_fn_u = pc_fn
 
             def pc_fn(u, b):
                 u, = tuple(iter_sub(u))
                 pc_fn_u(u, b)
 
-        if b.is_packed:
+        if b_packed.is_packed:
             pc_fn_b = pc_fn
 
             def pc_fn(u, b):