From d0f887c9045960b59ae40bb7d77f3f55cbb9ed02 Mon Sep 17 00:00:00 2001 From: Mikolaj Konarski Date: Tue, 13 Jan 2026 12:11:18 +0100 Subject: Let X.fromListOuterSN and ravelOuterN take NonEmpty --- src/Data/Array/Nested/Mixed.hs | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'src/Data/Array/Nested/Mixed.hs') diff --git a/src/Data/Array/Nested/Mixed.hs b/src/Data/Array/Nested/Mixed.hs index b0d32db..f0dadb4 100644 --- a/src/Data/Array/Nested/Mixed.hs +++ b/src/Data/Array/Nested/Mixed.hs @@ -28,7 +28,6 @@ import Control.Monad.ST import Data.Array.RankedS qualified as S import Data.Bifunctor (bimap) import Data.Coerce -import Data.Foldable (toList) import Data.Int import Data.Kind (Type) import Data.List.NonEmpty (NonEmpty(..)) @@ -423,7 +422,7 @@ instance Storable a => Elt (Primitive a) where mscalar (Primitive x) = M_Primitive ZSX (X.scalar x) mfromListOuterSN sn l@(arr1 :| _) = let sh = mshape arr1 - in M_Primitive (SKnown sn :$% sh) (X.fromListOuterSN sn sh (map (\(M_Primitive _ a) -> a) (toList l))) + in M_Primitive (SKnown sn :$% sh) (X.fromListOuterSN sn sh ((\(M_Primitive _ a) -> a) <$> l)) mtoListOuter (M_Primitive sh arr) = map (M_Primitive (shxTail sh)) (X.toListOuter arr) {-# INLINE mlift #-} -- cgit v1.2.3-70-g09d2