No, this should be `dist_grid_shape = ()` -- look at the other cases below to see how they do it.  There are no distributed dimensions, so this case is trivial.  `comm_size` may be an arbitrary integer, so the dimensionality of `(1,) * comm_size` would be wrong, in general.