Format and fix test bug with new idealmaps signature

Author: ptiede

examples/advanced/FitPS/main.jl

@@ -143,7 +143,7 @@ post = VLBIPosterior(skym, dlcamp, dcphase)
 # We also need to import Enzyme to allow for automatic differentiation.
 using Optimization, OptimizationLBFGSB
 # tpost = asflat(post)
-xopt, sol = comrade_opt(post, LBFGSB(); initial_params=prior_sample(rng, post), maxiters = 5000)
+xopt, sol = comrade_opt(post, LBFGSB(); initial_params = prior_sample(rng, post), maxiters = 5000)
 
 using CairoMakie
 using DisplayAs #hide

examples/intermediate/StokesIImaging/main.jl

@@ -231,7 +231,7 @@ chain = sample(rng, post, NUTS(0.8), 700; n_adapts = 500, initial_params = xopt)
 
 
 # Now we prune the adaptation phase
-chain = chain[begin+500:end]
+chain = chain[(begin + 500):end]
 
 #-
 # !!! warning

src/posterior/vlbiposterior.jl

@@ -137,7 +137,7 @@ If `add_thermal_noise` is true then baseline based thermal noise is added. Other
 return the model visibilities.
 """
 function simulate_observation(rng::Random.AbstractRNG, post::VLBIPosterior, θ; add_thermal_noise = true)
-    v0 = forward_model(post, θ)
+    v0 = last(forward_model(post, θ))
     Σn = _visnoise(first(post.data))
     data = post.data
 

test/Core/models.jl

@@ -150,11 +150,11 @@ end
         @test tt.dynamic == Comrade.skymodel(oskym, x.dynamic)
         @test tt.static == Comrade.skymodel(oskyf, x.static)
 
-        vtot = last(Comrade.idealmaps(oskytot, (; sky = x)))
-        vdyn = last(Comrade.idealmaps(oskym, (; sky = x.dynamic)))
-        vstat = last(Comrade.idealmaps(oskyf, (; sky = x.static)))
+        vtot = last(Comrade.idealmaps(Comrade.VisData(), oskytot, (; sky = x)))
+        vdyn = last(Comrade.idealmaps(Comrade.VisData(), oskym, (; sky = x.dynamic)))
+        vstat = last(Comrade.idealmaps(Comrade.VisData(), oskyf, (; sky = x.static)))
 
-        @test_opt Comrade.idealmaps(oskytot, (; sky = x))
+        @test_opt Comrade.idealmaps(Comrade.VisData(), oskytot, (; sky = x))
 
         @test vtot ≈ vdyn + vstat
     end