fix split_bases and split_indices for ElementVector

kinnala · Jan 10, 2024 · 246cf65 · 246cf65
1 parent 58f9a88
commit 246cf65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skfem/assembly/basis/abstract_basis.py b/skfem/assembly/basis/abstract_basis.py
@@ -338,7 +338,7 @@ def split_indices(self) -> List[ndarray]:
                                e.interior_dofs])
             return output
         elif isinstance(self.elem, ElementVector):
-            ndims = self.mesh.dim()
+            ndims = self.elem.dim
             e = self.elem.elem
             for k in range(ndims):
                 output.append(np.concatenate((
@@ -359,7 +359,7 @@ def split_bases(self) -> List['AbstractBasis']:
         elif isinstance(self.elem, ElementVector):
             return [type(self)(self.mesh, self.elem.elem, self.mapping,
                                quadrature=self.quadrature)
-                    for _ in range(self.mesh.dim())]
+                    for _ in range(self.elem.dim)]
         return [self]
 
     @property