aboutsummaryrefslogtreecommitdiff
path: root/src/Data
diff options
context:
space:
mode:
authorMikolaj Konarski <mikolaj.konarski@funktory.com>2025-12-11 14:22:07 +0100
committerMikolaj Konarski <mikolaj.konarski@funktory.com>2026-01-31 10:53:04 +0100
commit3936da7066be2ed4c0e43f7aed5a4d3cf019fdce (patch)
tree0540e2df8542a95840150607bf596e657bf705f8 /src/Data
parentbf00298304d866da1ace4878a46e77e81613284a (diff)
Generalize shsTakeIx
Diffstat (limited to 'src/Data')
-rw-r--r--src/Data/Array/Nested/Shaped.hs2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/Data/Array/Nested/Shaped.hs b/src/Data/Array/Nested/Shaped.hs
index 99ad590..d23a025 100644
--- a/src/Data/Array/Nested/Shaped.hs
+++ b/src/Data/Array/Nested/Shaped.hs
@@ -56,7 +56,7 @@ ssize = shsSize . sshape
sindex :: Elt a => Shaped sh a -> IIxS sh -> a
sindex (Shaped arr) idx = mindex arr (ixxFromIxS idx)
-shsTakeIx :: Proxy sh' -> ShS (sh ++ sh') -> IIxS sh -> ShS sh
+shsTakeIx :: Proxy sh' -> ShS (sh ++ sh') -> IxS sh i -> ShS sh
shsTakeIx _ _ ZIS = ZSS
shsTakeIx p sh (_ :.$ idx) = case sh of n :$$ sh' -> n :$$ shsTakeIx p sh' idx