From b492dc49e7dbffb5b336ae1e642b7e6f2b33708d Mon Sep 17 00:00:00 2001
From: kmdeck <kdeck@caltech.edu>
Date: Fri, 12 Jul 2024 13:39:33 -0700
Subject: [PATCH] point space support

---
 src/MatrixFields/matrix_shape.jl | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/MatrixFields/matrix_shape.jl b/src/MatrixFields/matrix_shape.jl
index 21434fc83c..f59f2113f0 100644
--- 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()