point space support

CliMA · Jul 26, 2024 · b492dc4 · b492dc4
1 parent cec368e
commit b492dc4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/MatrixFields/matrix_shape.jl b/src/MatrixFields/matrix_shape.jl
@@ -22,12 +22,11 @@ matrix_shape(matrix_field, matrix_space) = _matrix_shape(
     matrix_space.staggering,
 )
 
-function matrix_shape(matrix_field, matrix_space::Spaces.AbstractSpectralElementSpace)
+function matrix_shape(matrix_field, matrix_space::Union{Spaces.AbstractSpectralElementSpace, Spaces.PointSpace})
     @assert eltype(matrix_field) <: DiagonalMatrixRow
     Square()
 end
 
-
 _matrix_shape(::Type{Int}, _) = Square()
 _matrix_shape(::Type{PlusHalf{Int}}, ::Spaces.CellCenter) = FaceToCenter()
 _matrix_shape(::Type{PlusHalf{Int}}, ::Spaces.CellFace) = CenterToFace()