I have an open PR that changes `LocalArray.shape` to `LocalArray.global_shape`.