fix PowerManifold

JuliaManifolds · Aug 27, 2024 · 682717d · 682717d
1 parent cc35541
commit 682717d
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/PowerManifold.jl b/src/PowerManifold.jl
@@ -1319,7 +1319,7 @@ end
 
 Base.@propagate_inbounds @inline function _read(
     M::AbstractPowerManifold,
-    rep_size::Tuple,
+    rep_size::Union{Tuple,Nothing},
     x::AbstractArray,
     i::Int,
 )
@@ -1328,7 +1328,7 @@ end
 
 Base.@propagate_inbounds @inline function _read(
     ::Union{PowerManifoldNested,PowerManifoldNestedReplacing},
-    rep_size::Tuple,
+    rep_size::Union{Tuple,Nothing},
     x::AbstractArray,
     i::Tuple,
 )
@@ -1765,15 +1765,25 @@ function Weingarten!(M::AbstractPowerManifold, Y, p, X, V)
     return Y
 end
 
-@inline function _write(M::AbstractPowerManifold, rep_size::Tuple, x::AbstractArray, i::Int)
+@inline function _write(
+    M::AbstractPowerManifold,
+    rep_size::Union{Tuple,Nothing},
+    x::AbstractArray,
+    i::Int,
+)
     return _write(M, rep_size, x, (i,))
 end
 
 @inline function _is_nested_write_getindex(::PowerManifoldNested, x)
     return !isbitstype(eltype(x))
 end
 
-@inline function _write(M::PowerManifoldNested, ::Tuple, x::AbstractArray, i::Tuple)
+@inline function _write(
+    M::PowerManifoldNested,
+    ::Union{Tuple,Nothing},
+    x::AbstractArray,
+    i::Tuple,
+)
     if _is_nested_write_getindex(M, x)
         return x[i...]
     else