Keep coupled evidence audit linear in state size and repeat raw timing

experiments/check_coupled_trial.py

@@ -79,6 +79,7 @@ def inspect_case_artifacts(folder):
     check('base_physical_normalization',near_vector(x,[v*a for v,a in zip(physical,sf)],norm(x)))
     layout=rows(folder/(P+'layout.csv'));vl=layout_for(layout,'value',size);rl=layout_for(layout,'residual',size)
     gi=[i for name,(start,n) in vl.items() if name in GRAVITY_VALUES for i in range(start,start+n)]
+    gravity_value_indices=set(gi)
     gr=[i for name,(start,n) in rl.items() if name in GRAVITY_ROWS for i in range(start,start+n)]
     f0,pn,jn=norm(f),norm(p),norm(j)
     check('finite_positive_base_residual',math.isfinite(f0) and f0>0)
@@ -159,7 +160,7 @@ def inspect_case_artifacts(folder):
             gc=get('gravity_correction'+suffix,save_gravity)
             if save_gravity:
                 check(label+'gravity_partition_present',bool(gi) and bool(gr))
-                check(label+'gravity_only_correction',len(gc)==size and all(v==0 for k,v in enumerate(gc) if k not in set(gi)))
+                check(label+'gravity_only_correction',len(gc)==size and all(v==0 for k,v in enumerate(gc) if k not in gravity_value_indices))
                 if flags['before_prepared']:
                     check(label+'gravity_displacement',near_vector(vectors['before'][0],[v+w for v,w in zip(raw_state,gc)],norm(raw_state)+norm(gc)))
                     target=norm([vectors['before'][1][k]-prediction[k] for k in gr])