Same as above but with `axis='observation'`