diff --git a/skfem/assembly/basis/abstract_basis.py b/skfem/assembly/basis/abstract_basis.py index 8b21baa7..1a1c7962 100644 --- 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