diff options
| author | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-01-13 12:11:18 +0100 |
|---|---|---|
| committer | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-02-12 10:16:29 +0100 |
| commit | 40408abdcaa5336ef72dbac03bbebfc42677d80b (patch) | |
| tree | 392aca0465aa79d1e5cd6cc30df8c4106dcb7c7a /src/Data/Array/Nested | |
| parent | cd3baf1c529434f8d917a9eb74c5710fef480cdc (diff) | |
Let X.fromListOuterSN and ravelOuterN take NonEmpty
Diffstat (limited to 'src/Data/Array/Nested')
| -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 9a1e054..8b46b6e 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(..)) @@ -420,7 +419,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 #-} |