Whoops! Sorry, I was looking at the `Product` network, which should also not use the `kwarg` dimensions. I like `ens_dimensions` too.