Allow list input

patnr · Sep 29, 2023 · 6ca6a4c · 6ca6a4c
1 parent 6a4d065
commit 6ca6a4c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/TPFA_ResSim/plotting.py b/TPFA_ResSim/plotting.py
@@ -74,6 +74,7 @@ def plt_field(self, ax, Z, style="default", wells=True,
             raise ValueError(f"Unsupported coord_type: {coord_type}")
 
         # Apply transform
+        Z = np.asarray(Z)
         Z = kwargs.pop("transf")(Z)
 
         # Need to transpose coz orientation is model.shape==(Nx, Ny),