lint

india_forecast_app/data/nwp.py

@@ -8,8 +8,6 @@
 def regrid_nwp_data(nwp_ds: xr.Dataset, target_coords_path: str) -> xr.Dataset:
     """This function loads the  NWP data, then regrids and saves it back out if the data is not
     on the same grid as expected. The data is resaved in-place.
-
-    method can be 'conservative' or 'bilinear'
     """
 
     logger.info(f"Regridding NWP data to expected grid to {target_coords_path}")
@@ -27,7 +25,7 @@ def regrid_nwp_data(nwp_ds: xr.Dataset, target_coords_path: str) -> xr.Dataset:
 
     if not needs_regridding:
         logger.info(f"No NWP regridding required - skipping this step")
-        return
+        return ds_raw
 
     # flip latitude, so its in ascending order
     if ds_raw.latitude[0] > ds_raw.latitude[-1]:

india_forecast_app/models/pvnet/utils.py

@@ -9,12 +9,11 @@
 import xarray as xr
 import yaml
 from ocf_datapipes.batch import BatchKey
-from ocf_datapipes.utils.consts import ELEVATION_MEAN, ELEVATION_STD
 from ocf_datapipes.config.model import NWP
+from ocf_datapipes.utils.consts import ELEVATION_MEAN, ELEVATION_STD
 from pydantic import BaseModel
 
 from india_forecast_app.data.nwp import regrid_nwp_data
-
 from .consts import (
     nwp_ecmwf_path,
     nwp_gfs_path,