From 46e48ad854f4c5f47b1a4dbf4059bf59e0bea7df Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Fri, 29 Nov 2024 16:15:23 +0100 Subject: Improve scastToMixed's implementation --- src/Data/Array/Nested/Internal/Shaped.hs | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'src/Data/Array/Nested') diff --git a/src/Data/Array/Nested/Internal/Shaped.hs b/src/Data/Array/Nested/Internal/Shaped.hs index 76e0e6d..2a8c206 100644 --- a/src/Data/Array/Nested/Internal/Shaped.hs +++ b/src/Data/Array/Nested/Internal/Shaped.hs @@ -488,8 +488,8 @@ stoMixed (Shaped arr) = arr -- compatibility check. scastToMixed :: forall sh sh' a. (Elt a, Rank sh ~ Rank sh') => IShX sh' -> Shaped sh a -> Mixed sh' a -scastToMixed shx (Shaped arr) - | Refl <- lemAppNil @sh' - , Refl <- lemAppNil @(MapJust sh) - , Refl <- lemRankMapJust (shCvtXS' @sh (mshape arr)) - = mcast (ssxFromShape (mshape arr)) shx (Proxy @'[]) arr +scastToMixed shx (Shaped arr) = + case shxEqual shx (mshape arr) of + Just Refl -> arr + Nothing -> error $ "scastToMixed: Given shape (" ++ show shx ++ ") is not equal to " ++ + "the shape of the array (" ++ show (mshape arr) ++ ")" -- cgit v1.2.3-70-g09d2