fix issues from update to ufl (deleting brackets!)

gusto/coordinates.py

@@ -99,7 +99,7 @@ def register_space(self, domain, space_name):
         space = domain.spaces(space_name)
 
         # Use the appropriate scalar function space if the space is vector
-        if np.prod(space.ufl_element().value_shape()) > 1:
+        if np.prod(space.ufl_element().value_shape) > 1:
             # TODO: get scalar space, and only compute coordinates if necessary
             logger.warning(f'Space {space_name} has more than one dimension, '
                            + 'and coordinates used for netCDF output have not '

gusto/limiters.py

@@ -44,7 +44,7 @@ def __init__(self, space, subspace=None):
 
         # check that space is DG1
         degree = space.ufl_element().degree()
-        if (space.ufl_element().sobolev_space().name != 'L2'
+        if (space.ufl_element().sobolev_space.name != 'L2'
             or ((type(degree) is tuple and np.any([deg != 1 for deg in degree]))
                 and degree != 1)):
             raise ValueError('DG1 limiter can only be applied to DG1 space')
@@ -111,7 +111,7 @@ def __init__(self, space):
             raise ValueError('The Theta Limiter can only be used on an extruded mesh')
 
         # check that horizontal degree is 1 and vertical degree is 2
-        sub_elements = space.ufl_element().sub_elements()
+        sub_elements = space.ufl_element().sub_elements
         if (sub_elements[0].family() not in ['Discontinuous Lagrange', 'DQ']
                 or sub_elements[1].family() != 'Lagrange'
                 or space.ufl_element().degree() != (1, 2)):

gusto/preconditioners.py

@@ -72,7 +72,7 @@ def initialize(self, pc):
         for i, Vi in enumerate(V):
 
             # Vector-valued spaces will have a non-empty value_shape
-            if Vi.ufl_element().value_shape():
+            if Vi.ufl_element().value_shape:
                 self.vidx = i
             else:
                 self.pidx = i
@@ -88,11 +88,11 @@ def initialize(self, pc):
         deg, _ = Vv.ufl_element().degree()
 
         # Assumes a tensor product cell (quads, triangular-prisms, cubes)
-        if not isinstance(Vp.ufl_element().cell(), TensorProductCell):
+        if not isinstance(Vp.ufl_element().cell, TensorProductCell):
             raise NotImplementedError("Currently only implemented for tensor product discretizations")
 
         # Only want the horizontal cell
-        cell, _ = Vp.ufl_element().cell()._cells
+        cell, _ = Vp.ufl_element().cell._cells
 
         DG = FiniteElement("DG", cell, deg)
         CG = FiniteElement("CG", interval, 1)

gusto/recovery/recovery.py

@@ -332,20 +332,20 @@ def find_eff_coords(V0):
     vec_DG1 = VectorFunctionSpace(mesh, DG1_element)
     x = SpatialCoordinate(mesh)
 
-    if isinstance(V0.ufl_element(), VectorElement) or V0.ufl_element().value_size() > 1:
+    if isinstance(V0.ufl_element(), VectorElement) or V0.ufl_element().value_size > 1:
         eff_coords_list = []
         V0_coords_list = []
 
         # treat this separately for each component
-        for i in range(V0.ufl_element().value_size()):
+        for i in range(V0.ufl_element().value_size):
             # fill an d-dimensional list with i-th coordinate
-            x_list = [x[i] for j in range(V0.ufl_element().value_size())]
+            x_list = [x[i] for j in range(V0.ufl_element().value_size)]
 
             # the i-th element in V0_coords_list is a vector with all components the i-th coord
             ith_V0_coords = Function(V0).project(as_vector(x_list))
             V0_coords_list.append(ith_V0_coords)
 
-        for i in range(V0.ufl_element().value_size()):
+        for i in range(V0.ufl_element().value_size):
             # slice through V0_coords_list to obtain the coords of the DOFs for that component
             x_list = [V0_coords[i] for V0_coords in V0_coords_list]
 

gusto/wrappers.py

@@ -255,7 +255,7 @@ def is_cg(V):
     elif type(ele) == VectorElement:
         return all([e.sobolev_space().name == "H1" for e in ele._sub_elements])
     else:
-        return V.ufl_element().sobolev_space().name == "H1"
+        return V.ufl_element().sobolev_space.name == "H1"
 
 
 class SUPGWrapper(Wrapper):
@@ -293,7 +293,7 @@ def setup(self):
             if is_cg(self.function_space):
                 vals = default_vals
             else:
-                space = self.function_space.ufl_element().sobolev_space()
+                space = self.function_space.ufl_element().sobolev_space
                 if space.name in ["HDiv", "DirectionalH"]:
                     vals = [default_vals[i] if space[i].name == "H1"
                             else 0. for i in range(dim)]