Change lwp_offset to tuple

Author: tukiains

mwrpy/level2/lev2_collocated.py

@@ -13,7 +13,7 @@ def generate_lev2_single(
     site: str | None,
     mwr_l1c_file: str | PathLike,
     output_file: str | PathLike,
-    lwp_offset: list[float | None] = [None, None],
+    lwp_offset: tuple[float | None, float | None] = (None, None),
     coeff_files: Sequence[str | PathLike] | None = None,
 ):
     with (
@@ -160,7 +160,7 @@ def generate_lev2_single(
                         hum_file=abs_hum_file.name
                         if prod in ("2P04", "2P07", "2P08")
                         else None,
-                        lwp_offset=[None, None],
+                        lwp_offset=(None, None),
                         coeff_files=coeff_files,
                     )
                     with netCDF4.Dataset(stability_file.name, "r") as nc_sta:
@@ -188,7 +188,7 @@ def generate_lev2_lhumpro(
     site: str | None,
     mwr_l1c_file: str | PathLike,
     output_file: str | PathLike,
-    lwp_offset: list[float | None] = [None, None],
+    lwp_offset: tuple[float | None, float | None] = (None, None),
     coeff_files: Sequence[str | PathLike] | None = None,
 ):
     with (
@@ -307,7 +307,7 @@ def generate_lev2_multi(
                 if prod not in ("2P02", "2P03")
                 else None,
                 hum_file=abs_hum_file.name if prod not in ("2P02", "2P03") else None,
-                lwp_offset=[None, None],
+                lwp_offset=(None, None),
                 coeff_files=coeff_files,
             )
 

mwrpy/level2/lwp_offset.py

@@ -13,7 +13,7 @@ def correct_lwp_offset(
     lwp_org: np.ndarray,
     index: np.ndarray,
     qf: np.ndarray,
-    offset_xd: list[float | None] = [None, None],
+    offset_xd: tuple[float | None, float | None] = (None, None),
 ) -> tuple[np.ndarray, np.ndarray]:
     """This function corrects Lwp offset using the
     2min Lwp standard deviation and the water vapor

mwrpy/level2/write_lev2_nc.py

@@ -31,7 +31,7 @@ def lev2_to_nc(
     site: str | None = None,
     temp_file: str | PathLike | None = None,
     hum_file: str | PathLike | None = None,
-    lwp_offset: list[float | None] = [None, None],
+    lwp_offset: tuple[float | None, float | None] = (None, None),
     coeff_files: Sequence[str | PathLike] | None = None,
 ):
     """This function reads Level 1 files,
@@ -45,7 +45,7 @@ def lev2_to_nc(
         site: Name of site.
         temp_file: Name of temperature product file.
         hum_file: Name of humidity product file.
-        lwp_offset: Offset for LWP correction.
+        lwp_offset: LWP offset with the previous day's last and next day's first reliable values.
         coeff_files: List of coefficient files.
 
     """
@@ -93,7 +93,7 @@ def get_products(
     coeff_files: Sequence[str | PathLike] | None,
     temp_file: str | PathLike | None = None,
     hum_file: str | PathLike | None = None,
-    lwp_offset: list[float | None] = [None, None],
+    lwp_offset: tuple[float | None, float | None] = (None, None),
 ) -> tuple[dict, dict, np.ndarray, np.ndarray]:
     """Derive specified Level 2 products."""
     lev1 = {key: value[:] for key, value in nclev1.variables.items()}

mwrpy/process_mwrpy.py

@@ -164,6 +164,7 @@ def process_product(prod: str, date: datetime.date, site: str):
                 lwp_offset[1] = csv_off.loc[
                     csv_off["date"] == xday[1].strftime("%m-%d"), "offset"
                 ].values[0]
+    lwp_offset_tuple = (lwp_offset[0], lwp_offset[1])
 
     itype = _read_site_config_yaml(site)["type"]
     if prod[0] == "1":
@@ -191,15 +192,15 @@ def process_product(prod: str, date: datetime.date, site: str):
             site=site,
             temp_file=temp_file,
             hum_file=hum_file,
-            lwp_offset=lwp_offset,
+            lwp_offset=lwp_offset_tuple,
         )
     elif prod == "single" and itype != "lhumpro_u90":
         generate_lev2_single(
-            site, _get_filename("1C01", date, site), output_file, lwp_offset
+            site, _get_filename("1C01", date, site), output_file, lwp_offset_tuple
         )
     elif itype == "lhumpro_u90":
         generate_lev2_lhumpro(
-            site, _get_filename("1C01", date, site), output_file, lwp_offset
+            site, _get_filename("1C01", date, site), output_file, lwp_offset_tuple
         )
     elif prod == "multi":
         generate_lev2_multi(site, _get_filename("1C01", date, site), output_file)