🐛[BUG]: Incorrect shift parameter in torch.roll inside forward()

[mhr123456789]

Version

1.3.0

On which installation method(s) does this occur?

No response

Describe the issue

Hi, I found a bug in the forward method.

In line 299, the code currently uses:

shifted_x = torch.roll(x, shifts=(-shift_lat, -shift_lat), dims=(1, 2))
It seems the second shift argument should be shift_lon instead of shift_lat:shifted_x = torch.roll(x, shifts=(-shift_lat, -shift_lon), dims=(1, 2))
This will cause issues when shift_lat != shift_lon (e.g. non-square windows).


### Minimum reproducible example

```shell

Relevant log output

Environment details

[coreyjadams]

Hi @mhr123456789 can you elaborate on where this is happening? A small reproducer would be very valuable.