Fix wrong-order keys in unflattened observations

This should just be gracefully handled, but was broken before.
The tests explicitly check that everything works by constructing
`OrderedDict`s with the "wrong" order.

Also, run flake8 and copy the CI's yapf output (which differs from my local yapf output).