fix some ComponentArray grads reverting to Vector

Author: marius311

src/autodiff.jl

@@ -1,6 +1,7 @@
 
 # this does basis promotion, unlike Zygote's default for AbstractArrays
 Zygote.accum(a::Field, b::Field) = a+b
+Zygote.accum(a::FieldTuple, b::FieldTuple) = Zygote.accum.(a,b)
 # this may create a LazyBinaryOp, unlike Zygote's
 Zygote.accum(a::FieldOp, b::FieldOp) = a+b
 

src/dataset.jl

@@ -230,7 +230,7 @@ function load_sim(;
         @warn "`rfid` will be removed in a future version. Use `fiducial_θ=(r=...,)` instead."
         fiducial_θ = merge(fiducial_θ,(r=rfid,))
     end
-    Aϕ₀ = get(fiducial_θ, :Aϕ, 1)
+    Aϕ₀ = T(get(fiducial_θ, :Aϕ, 1))
     fiducial_θ = Base.structdiff(fiducial_θ, NamedTuple{(:Aϕ,)}) # remove Aϕ key if present
     if (Cℓ == nothing)
         Cℓ = camb(;fiducial_θ..., ℓmax=ℓmax)
@@ -241,7 +241,7 @@ function load_sim(;
             error("ℓmax of `Cℓ` argument should be higher than $ℓmax for this configuration.")
         end
     end
-    r₀ = Cℓ.params.r
+    r₀ = T(Cℓ.params.r)
 
     # noise Cℓs (these are non-debeamed, hence beamFWHM=0 below; the beam comes in via the B operator)
     if (Cℓn == nothing)
@@ -264,7 +264,7 @@ function load_sim(;
     Cf̃  = Cℓ_to_Cov(pol, proj, (Cℓ.total[k]           for k in ks)...)
     Cn̂  = Cℓ_to_Cov(pol, proj, (Cℓn[k]                for k in ks)...)
     if (Cn == nothing); Cn = Cn̂; end
-    Cf = ParamDependentOp((;r=r₀,   _...)->(Cfs + T(r/r₀)*Cft))
+    Cf = ParamDependentOp((;r=r₀,   _...)->(Cfs + (T(r)/r₀)*Cft))
     Cϕ = ParamDependentOp((;Aϕ=Aϕ₀, _...)->(T(Aϕ) * Cϕ₀))
 
     # data mask