diff options
| author | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-01-13 12:11:18 +0100 |
|---|---|---|
| committer | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-01-13 12:11:18 +0100 |
| commit | d0f887c9045960b59ae40bb7d77f3f55cbb9ed02 (patch) | |
| tree | 3de1d53bc941e8e798e970195faea96a3298236d /src/Data/Array/Nested/Mixed.hs | |
| parent | 6959b7e4769289983f008d558a71fe0dd2e3d279 (diff) | |
Let X.fromListOuterSN and ravelOuterN take NonEmpty
Diffstat (limited to 'src/Data/Array/Nested/Mixed.hs')
| -rw-r--r-- | src/Data/Array/Nested/Mixed.hs | 3 |
1 files changed, 1 insertions, 2 deletions
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 #-} |