re order dims

Author: peterdudfield

nowcasting_dataset/data_sources/satellite/satellite_model.py

@@ -42,6 +42,10 @@ def model_validation(cls, v):
         v.check_data_var_dim(v.x_geostationary, ("example", "x_geostationary_index"))
         v.check_data_var_dim(v.y_geostationary, ("example", "y_geostationary_index"))
 
+        # re-order dims
+        if v.data.dims != satellite_expected_dims_order:
+            v.__setitem__("data", v.data.transpose(*satellite_expected_dims_order))
+
         return v