Tweak plotting

util/sugar.py

@@ -51,14 +51,14 @@ def save_state(sky, pos, pre, name, samples):
     varlim = [0., np.max(rel.val)]
     for ii in range(7):
         fi = ift.DomainTupleFieldInserter(dom, 0, (int(ii*24/dt),)).adjoint
-        p.add(fi(sc.mean), zmin=vlim[0], zmax=vlim[1])
+        p.add(fi(sc.mean), zmin=vlim[0], zmax=vlim[1], title=f'Day {ii}, Posterior Mean')
     for ii in range(7):
         fi = ift.DomainTupleFieldInserter(dom, 0, (int(ii*24/dt),)).adjoint
-        p.add(fi(rel), zmin=varlim[0], zmax=varlim[1])
+        p.add(fi(rel), zmin=varlim[0], zmax=varlim[1], title=f'Day {ii}, Rel. Stddev')
     for ii in range(7):
         fi = ift.DomainTupleFieldInserter(dom, 0, (int(ii*24/dt),)).adjoint
-        p.add(fi(sk), zmin=vlim[0], zmax=vlim[1])
-    p.output(nx=7, ny=3, name=f'{pre}{name}.png', xsize=20, ysize=8)
+        p.add(fi(sk), zmin=vlim[0], zmax=vlim[1], title=f'Day {ii}, Latent Mean')
+    p.output(nx=7, ny=3, name=f'{pre}{name}.png', xsize=25, ysize=10)
 
 
 def baselines(alst):