climdata 0.1.1__py2.py3-none-any.whl → 0.1.2__py2.py3-none-any.whl

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/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/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.dist-info → climdata-0.1.2.dist-info}/METADATA

@@ -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.dist-info → climdata-0.1.2.dist-info}/RECORD

@@ -1,20 +1,20 @@
-climdata/__init__.py,sha256=rR9zjk6GF4r-_jMpBDYi_kNis5lNNu_IOIottOLRcgU,462
+climdata/__init__.py,sha256=NkkQIZAYPwWJJCxQOfGoBF7fSuMAt2a6TVrEVXau8Ac,462
 climdata/__main__.py,sha256=Gn-CeD1_A0xSU8lvpuDJeniNtVgkwDpcRhxBOlrfV_w,82
 climdata/main.py,sha256=4_tm82v6tEa1fH1IeL2IslyjiB9NWOk46A-_QcsHg64,1586
 climdata/conf/config.yaml,sha256=_WHG5LDjzV28ZeyGVglnbXBmrAlPx0pV-5q-OhCYE3o,954
 climdata/conf/mappings/parameters.yaml,sha256=HzTjJnEWEYI4PiW29cGp0UbRBDc1f_NhgJdlYT6MXNY,4100
 climdata/conf/mappings/variables.yaml,sha256=mO13rtF0XOzP4fjU5oyao44GmRM0Jrr1RjUBBwg04IU,2030
-climdata/datasets/CMIPCloud.py,sha256=kriuH_B2GZLOH8vA6B3soqfAj5-iiiyYyId6l7vmMnk,5509
+climdata/datasets/CMIPCloud.py,sha256=uIIIyZWrgValIJWlDxhwNjTSHLlT6KD0J4dPn-Qw_Qs,5536
 climdata/datasets/CMIPlocal.py,sha256=mEBJZOfWu8fNlAQksXAbslWL0sJYQfu_UA2zPuetsBk,8815
 climdata/datasets/DWD.py,sha256=DwqBBkRLE_FXyjZX38iVv3cduiO6uQc8MQaTvHQqxjA,2850
 climdata/datasets/ERA5.py,sha256=OTCCstF8j-T8VL3oO7pISgcybXCLMghv-fRBv9_dmvY,11251
-climdata/datasets/MSWX.py,sha256=rBA8k6KwidOBXjb2fAlUTJV7dPPZ8ewWMlA4uQvbGDo,9993
+climdata/datasets/MSWX.py,sha256=r7gGeXp3NLsfI4T9h9k4T1b6brY7BqBKpP-uwfunTHs,10516
 climdata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 climdata/utils/config.py,sha256=UhbCH1PEVkEIdLVgmEEvb39Tq5XX4fsNe5gN4B5WL9Y,1054
 climdata/utils/utils_download.py,sha256=lCMFIzzZHVptLwpwCslbQhhie5mWsKzphsz_SQ3kHo0,33356
-climdata-0.1.1.dist-info/licenses/LICENSE,sha256=f_3cGg8TC4V7GSbVaJo9b_hy-iY3q_ZpWq1MN2VQcnw,1076
-climdata-0.1.1.dist-info/METADATA,sha256=46WXCUH8iKpdbGeFhPizyMBanAWLnXI-VIhG8hlwcrY,8357
-climdata-0.1.1.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
-climdata-0.1.1.dist-info/entry_points.txt,sha256=tiYPawJoZiBj7lU67vNTCb3esSvx_d-lBvOffA26ouU,47
-climdata-0.1.1.dist-info/top_level.txt,sha256=BPnAhRqg8vk580nSJDXTdLmfq6OZ_LR8eNTgrRabArw,9
-climdata-0.1.1.dist-info/RECORD,,
+climdata-0.1.2.dist-info/licenses/LICENSE,sha256=f_3cGg8TC4V7GSbVaJo9b_hy-iY3q_ZpWq1MN2VQcnw,1076
+climdata-0.1.2.dist-info/METADATA,sha256=F_W10wm162jJFWNcWTpXTVvxLmmq4dPfy1-omDLZ8vc,8357
+climdata-0.1.2.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
+climdata-0.1.2.dist-info/entry_points.txt,sha256=tiYPawJoZiBj7lU67vNTCb3esSvx_d-lBvOffA26ouU,47
+climdata-0.1.2.dist-info/top_level.txt,sha256=BPnAhRqg8vk580nSJDXTdLmfq6OZ_LR8eNTgrRabArw,9
+climdata-0.1.2.dist-info/RECORD,,