From 64c47540bdcf764f6ac3644cabac00e9605c3fa7 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Mon, 30 Jun 2025 10:44:10 +0200 Subject: Fix rerank docs --- src/Data/Array/Nested/Mixed.hs | 2 +- src/Data/Array/Nested/Ranked.hs | 2 +- src/Data/Array/Nested/Shaped.hs | 1 + 3 files changed, 3 insertions(+), 2 deletions(-) (limited to 'src/Data/Array/Nested') diff --git a/src/Data/Array/Nested/Mixed.hs b/src/Data/Array/Nested/Mixed.hs index f59d58e..144230e 100644 --- a/src/Data/Array/Nested/Mixed.hs +++ b/src/Data/Array/Nested/Mixed.hs @@ -833,7 +833,7 @@ mrerankP ssh sh2 f (M_Primitive sh arr) = (\a -> let M_Primitive _ r = f (M_Primitive sh1 a) in r) arr) --- | See the caveats at @X.rerank@. +-- | See the caveats at 'Data.Array.XArray.rerank'. mrerank :: forall sh1 sh2 sh a b. (PrimElt a, PrimElt b) => StaticShX sh -> IShX sh2 -> (Mixed sh1 a -> Mixed sh2 b) diff --git a/src/Data/Array/Nested/Ranked.hs b/src/Data/Array/Nested/Ranked.hs index 97b4c7c..9778c54 100644 --- a/src/Data/Array/Nested/Ranked.hs +++ b/src/Data/Array/Nested/Ranked.hs @@ -224,7 +224,7 @@ rrerankP sn sh2 f (Ranked arr) -- then: -- -- @ --- rrerank _ _ _ f arr :: Ranked 5 Float +-- rrerank _ _ _ f arr :: Ranked 6 Float -- @ -- -- and this result will have shape @[3, 0, 4, 0, 0, 0]@. Note that the diff --git a/src/Data/Array/Nested/Shaped.hs b/src/Data/Array/Nested/Shaped.hs index 2b0b6b5..198a068 100644 --- a/src/Data/Array/Nested/Shaped.hs +++ b/src/Data/Array/Nested/Shaped.hs @@ -191,6 +191,7 @@ srerankP sh sh2 f sarr@(Shaped arr) (\a -> let Shaped r = f (Shaped a) in r) arr) +-- | See the caveats at 'Data.Array.XArray.rerank'. srerank :: forall sh1 sh2 sh a b. (PrimElt a, PrimElt b) => ShS sh -> ShS sh2 -> (Shaped sh1 a -> Shaped sh2 b) -- cgit v1.2.3-70-g09d2