aboutsummaryrefslogtreecommitdiff
path: root/src/Data/Array/Nested/Ranked.hs
diff options
context:
space:
mode:
authorMikolaj Konarski <mikolaj.konarski@funktory.com>2026-04-02 11:54:26 +0200
committerMikolaj Konarski <mikolaj.konarski@funktory.com>2026-04-02 11:54:26 +0200
commit086145dfee33e6e198abc148a6375ce929b2c88f (patch)
tree1837166ce6bdcff1554df29bceb34e4ec5796d19 /src/Data/Array/Nested/Ranked.hs
parent6e48137f2b7d9613f85599f13dbaa949045f96a6 (diff)
Audit remaining uses of KnownNat and SNat patterns
Diffstat (limited to 'src/Data/Array/Nested/Ranked.hs')
-rw-r--r--src/Data/Array/Nested/Ranked.hs8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/Data/Array/Nested/Ranked.hs b/src/Data/Array/Nested/Ranked.hs
index 26c9203..2d8b624 100644
--- a/src/Data/Array/Nested/Ranked.hs
+++ b/src/Data/Array/Nested/Ranked.hs
@@ -65,7 +65,7 @@ rindexPartial (Ranked arr) idx =
-- 'rgeneratePrim'.
rgenerate :: forall n a. KnownElt a => IShR n -> (IIxR n -> a) -> Ranked n a
rgenerate sh f
- | sn@SNat <- shrRank sh
+ | sn <- shrRank sh
, Dict <- lemKnownReplicate sn
, Refl <- lemRankReplicate sn
= Ranked (mgenerate (shxFromShR sh) (f . ixrFromIxX))
@@ -117,11 +117,11 @@ rsumAllPrim (Ranked arr) = msumAllPrim arr
rtranspose :: forall n a. Elt a => PermR -> Ranked n a -> Ranked n a
rtranspose perm arr
- | sn@SNat <- rrank arr
+ | sn <- rrank arr
, Dict <- lemKnownReplicate sn
- , length perm <= fromIntegral (natVal (Proxy @n))
+ , length perm <= fromSNat' sn
= rlift sn
- (\ssh' -> X.transposeUntyped (natSing @n) ssh' perm)
+ (\ssh' -> X.transposeUntyped sn ssh' perm)
arr
| otherwise
= error "Data.Array.Nested.rtranspose: Permutation longer than rank of array"