From 4560294d99714ec734009e4904965bd979ccd7ae Mon Sep 17 00:00:00 2001 From: Kamlesh Date: Tue, 7 Jan 2025 11:55:53 +0530 Subject: [PATCH] Update attribute name to fix the non-acceptance of cli kwargs for earthengine --- weather_mv/loader_pipeline/ee.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/weather_mv/loader_pipeline/ee.py b/weather_mv/loader_pipeline/ee.py index ebad6b3a..ef9389ba 100644 --- a/weather_mv/loader_pipeline/ee.py +++ b/weather_mv/loader_pipeline/ee.py @@ -461,13 +461,13 @@ class ConvertToAsset(beam.DoFn, KwargsFactoryMixin): Attributes: ee_asset_type: The type of asset to ingest in the earth engine. Default: IMAGE. asset_location: The bucket location at which asset files will be pushed. - open_dataset_kwargs: A dictionary of kwargs to pass to xr.open_dataset(). + xarray_open_dataset_kwargs: A dictionary of kwargs to pass to xr.open_dataset(). disable_grib_schema_normalization: A flag to turn grib schema normalization off; Default: on. """ asset_location: str ee_asset_type: str = 'IMAGE' - open_dataset_kwargs: t.Optional[t.Dict] = None + xarray_open_dataset_kwargs: t.Optional[t.Dict] = None disable_grib_schema_normalization: bool = False group_common_hypercubes: t.Optional[bool] = False band_names_dict: t.Optional[t.Dict] = None @@ -493,7 +493,7 @@ def convert_to_asset(self, queue: Queue, uri: str): job_start_time = get_utc_timestamp() with open_dataset(uri, - self.open_dataset_kwargs, + self.xarray_open_dataset_kwargs, self.disable_grib_schema_normalization, initialization_time_regex=self.initialization_time_regex, forecast_time_regex=self.forecast_time_regex,