climdata 0.1.1__tar.gz → 0.1.2__tar.gz

{climdata-0.1.1 → climdata-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: climdata
-Version: 0.1.1
+Version: 0.1.2
 Summary: This project automates the fetching and extraction of weather data from multiple sources — such as MSWX, DWD HYRAS, ERA5-Land, NASA-NEX-GDDP, and more — for a given location and time range.
 Author-email: Kaushik Muduchuru <kaushik.reddy.m@gmail.com>
 License: MIT License

{climdata-0.1.1 → climdata-0.1.2}/climdata/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Kaushik Muduchuru"""
 __email__ = "kaushik.reddy.m@gmail.com"
-__version__ = "0.1.1"
+__version__ = "0.1.2"
 
 from .utils.utils_download import * # etc.
 from .utils.config import load_config

{climdata-0.1.1 → climdata-0.1.2}/climdata/datasets/CMIPCloud.py

@@ -27,6 +27,7 @@ class CMIPCloud:
             if len(col_subset.df) == 0:
                 continue
             self.col_subsets.append(col_subset)
+            self.col = col
         return self.col_subsets
 
     def load(self):

{climdata-0.1.1 → climdata-0.1.2}/climdata/datasets/MSWX.py

@@ -160,6 +160,8 @@ class MSWXmirror:
             local_path = os.path.join(output_dir, provider, parameter_key, f)
             try:
                 ds = xr.open_dataset(local_path, chunks='auto', engine='netcdf4')[param_info.name]
+                # Rename DataArray to parameter_key
+                ds = ds.rename(parameter_key)
                 valid_dsets.append(ds)
             except Exception as e:
                 print(f"Skipping file due to error: {f}\n{e}")
@@ -167,7 +169,7 @@ class MSWXmirror:
         dset = xr.concat(valid_dsets, dim='time')
         dset = dset.transpose('time', 'lat', 'lon')
         self.dataset = self._fix_coords(dset)
-        return dset
+        return self.dataset
 
     def to_zarr(self, zarr_filename):
         if self.dataset is None:
@@ -227,10 +229,10 @@ class MSWXmirror:
                 ds_subset = ds.sel(lon=lon, lat=lat, method="nearest")
 
         elif box is not None:
-            min_lon, min_lat, max_lon, max_lat = box
+            # Accept dict: {'lat_min': ..., 'lat_max': ..., 'lon_min': ..., 'lon_max': ...}
             ds_subset = ds.sel(
-                lon=slice(min_lon, max_lon),
-                lat=slice(min_lat, max_lat)
+                lon=slice(box['lon_min'], box['lon_max']),
+                lat=slice(box['lat_min'], box['lat_max'])
             )
 
         elif shapefile is not None:
@@ -248,7 +250,8 @@ class MSWXmirror:
 
         else:
             raise ValueError("Must provide either point, box, or shapefile.")
-
+        self.dataset = ds_subset
+        self.dataset = self.dataset.to_dataset()
         return ds_subset
     
     def to_dataframe(self, ds=None):
@@ -287,6 +290,12 @@ class MSWXmirror:
             ds.name: "value"
         })
         return df
+    def save_netcdf(self, filename):
+        if self.dataset is not None:
+            if "time" in self.dataset.variables:
+                self.dataset["time"].encoding.clear()
+            self.dataset.to_netcdf(filename)
+            # print(f"Saved NetCDF to {filename}")
 
     def format(self, df):
         """
@@ -299,4 +308,4 @@ class MSWXmirror:
 
         df = df[["latitude", "longitude", "time", "source", "variable", "value", "units"]]
         return df
-    
+

{climdata-0.1.1 → climdata-0.1.2}/climdata.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: climdata
-Version: 0.1.1
+Version: 0.1.2
 Summary: This project automates the fetching and extraction of weather data from multiple sources — such as MSWX, DWD HYRAS, ERA5-Land, NASA-NEX-GDDP, and more — for a given location and time range.
 Author-email: Kaushik Muduchuru <kaushik.reddy.m@gmail.com>
 License: MIT License