xradio 0.0.43__py3-none-any.whl → 0.0.44__py3-none-any.whl

xradio/image/_util/_casacore/xds_from_casacore.py

@@ -174,15 +174,13 @@ def _casa_image_to_xds_attrs(img_full_path: str, history: bool = True) -> dict:
             "type": "sky_coord",
             "equinox": ap_equinox if ap_equinox else None,
             "value": [0.0, 0.0],
-            "cdelt": [0.0, 0.0],
             "units": ["rad", "rad"],
         }
-        for c, r in zip(["crval", "cdelt"], ["value", "cdelt"]):
-            for i in range(2):
-                unit = u.Unit(_get_unit(coord_dir_dict["units"][i]))
-                q = coord_dir_dict[c][i] * unit
-                x = q.to("rad")
-                dir_dict["reference"][r][i] = x.value
+        for i in range(2):
+            unit = u.Unit(_get_unit(coord_dir_dict["units"][i]))
+            q = coord_dir_dict["crval"][i] * unit
+            x = q.to("rad")
+            dir_dict["reference"]["value"][i] = x.value
         k = "latpole"
         if k in coord_dir_dict:
             for j in (k, "longpole"):

xradio/image/_util/_casacore/xds_to_casacore.py

@@ -24,7 +24,7 @@ def _compute_direction_dict(xds: xr.Dataset) -> dict:
     direction["projection_parameters"] = xds_dir["projection_parameters"]
     direction["units"] = np.array(xds_dir["reference"]["units"], dtype="<U16")
     direction["crval"] = np.array(xds_dir["reference"]["value"])
-    direction["cdelt"] = np.array(xds_dir["reference"]["cdelt"])
+    direction["cdelt"] = np.array((xds.l.cdelt, xds.m.cdelt))
     direction["crpix"] = _compute_sky_reference_pixel(xds)
     direction["pc"] = np.array(xds_dir["pc"])
     direction["axes"] = ["Right Ascension", "Declination"]

xradio/image/_util/_fits/xds_from_fits.py

@@ -176,7 +176,6 @@ def _xds_direction_attrs_from_header(helpers: dict, header) -> dict:
         "equinox": ref_eqx,
         "units": ["rad", "rad"],
         "value": [0.0, 0.0],
-        "cdelt": [0.0, 0.0],
     }
     dir_axes = helpers["dir_axes"]
     for i in dir_axes:
@@ -185,7 +184,6 @@ def _xds_direction_attrs_from_header(helpers: dict, header) -> dict:
         direction["reference"]["value"][i] = x.value
         x = helpers["cdelt"][i] * u.Unit(_get_unit(helpers["cunit"][i]))
         x = x.to("rad")
-        direction["reference"]["cdelt"][i] = x.value
     direction["latpole"] = {
         "value": header["LATPOLE"] * _deg_to_rad,
         "units": "rad",

xradio/image/_util/image_factory.py

@@ -68,7 +68,6 @@ def _add_common_attrs(
                 "equinox": "J2000",
                 "value": list(phase_center),
                 "units": ["rad", "rad"],
-                "cdelt": [-abs(cell_size[0]), abs(cell_size[1])],
             },
             "longpole": {"type": "quantity", "value": np.pi, "units": "rad"},
             "latpole": {"type": "quantity", "value": 0.0, "units": "rad"},

xradio/measurement_set/measurement_set_xds.py

@@ -47,7 +47,7 @@ class MeasurementSetXds(xr.Dataset):
             if "xds" in attrs_name:
                 del copy_cor_xds.attrs[attrs_name]
                 xr.Dataset.to_zarr(
-                    self.attrs[attrs_name], os.path.join(store, attrs_name, **kwargs)
+                    self.attrs[attrs_name], os.path.join(store, attrs_name), **kwargs
                 )
 
         # Save copy_cor_xds as zarr file.
@@ -78,10 +78,11 @@ class MeasurementSetXds(xr.Dataset):
         >>> # Select data group 'corrected' and polarization 'XX' using a dict.
         >>> selected_ms_xds = ms_xds.sel({'data_group_name':'corrected', 'polarization':'XX')
         """
+
         if "data_group_name" in indexers_kwargs:
             data_group_name = indexers_kwargs["data_group_name"]
             del indexers_kwargs["data_group_name"]
-        if (indexers is not None) and ("data_group_name" in indexers):
+        elif (indexers is not None) and ("data_group_name" in indexers):
             data_group_name = indexers["data_group_name"]
             del indexers["data_group_name"]
         else:
@@ -99,11 +100,17 @@ class MeasurementSetXds(xr.Dataset):
 
             data_variables_to_drop = list(set(data_variables_to_drop))
 
-            return MeasurementSetXds(
+            sel_ms_xds = MeasurementSetXds(
                 super()
                 .sel(indexers, method, tolerance, drop, **indexers_kwargs)
                 .drop_vars(data_variables_to_drop)
             )
+
+            sel_ms_xds.attrs["data_groups"] = {
+                data_group_name: self.attrs["data_groups"][data_group_name]
+            }
+
+            return sel_ms_xds
         else:
             return MeasurementSetXds(
                 super().sel(indexers, method, tolerance, drop, **indexers_kwargs)

xradio/measurement_set/processing_set.py

@@ -193,10 +193,10 @@ class ProcessingSet(dict):
 
         spw_ids = []
         freq_axis_list = []
-        frame = self.get(0).frequency.attrs["frame"]
+        frame = self.get(0).frequency.attrs["observer"]
         for ms_xds in self.values():
             assert (
-                frame == ms_xds.frequency.attrs["frame"]
+                frame == ms_xds.frequency.attrs["observer"]
             ), "Frequency reference frame not consistent in Processing Set."
             if ms_xds.frequency.attrs["spectral_window_id"] not in spw_ids:
                 spw_ids.append(ms_xds.frequency.attrs["spectral_window_id"])

{xradio-0.0.43.dist-info → xradio-0.0.44.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xradio
-Version: 0.0.43
+Version: 0.0.44
 Summary: Xarray Radio Astronomy Data IO
 Author-email: Jan-Willem Steeb <jsteeb@nrao.edu>
 License: BSD 3-Clause License

{xradio-0.0.43.dist-info → xradio-0.0.44.dist-info}/RECORD

@@ -12,13 +12,13 @@ xradio/image/_util/__init__.py,sha256=M9lxD1Gc7kv0ucDEDbjLRuIEuESev-IG8j9EaCKUAk
 xradio/image/_util/casacore.py,sha256=DmBTHUQ6870N5ARuFnYSfjZSLniJYgsjrsICUlCREYM,4234
 xradio/image/_util/common.py,sha256=gCVIq547PZMOM7urkdwPXyKcfstD-loJFcf6YGa8CGM,9039
 xradio/image/_util/fits.py,sha256=gyGm06fuCKqVGK7uv-ObvQNfFawUDsIOa_nQyklM3Aw,329
-xradio/image/_util/image_factory.py,sha256=-ZCDkCg25HMN4ubDxeBPv24HsEpYB5_zJYajgPx6_pg,10514
+xradio/image/_util/image_factory.py,sha256=9BwW1pDxv1XaJqbQKwO03kwQyXqNwtIfIyqabHATF5A,10448
 xradio/image/_util/zarr.py,sha256=lhQqVRC1GEWClG3zRbuDr2IlQBfXeDqaLUJIN-MVMxA,1652
 xradio/image/_util/_casacore/__init__.py,sha256=OlsiRE40o1jSbBI4khgQQzgfDYbAlOMKIhO4UFlbGhg,41
 xradio/image/_util/_casacore/common.py,sha256=ky999eTCWta8w-uIs-7P7rPhZRLuh9yTuQXAxPvaPm4,1579
-xradio/image/_util/_casacore/xds_from_casacore.py,sha256=o03KTPEyJpG78GJJeT2gqYZWe7gqqWQ9ned0UEQw710,42596
-xradio/image/_util/_casacore/xds_to_casacore.py,sha256=qSoKemfH4wX7-7QrEQa_y3Qitn7NCIoCV7fLfDhaaUM,15427
-xradio/image/_util/_fits/xds_from_fits.py,sha256=MNlZfW0exYBlCoMCA_Nd84eggNw_W8WhjSICgAihyLk,28543
+xradio/image/_util/_casacore/xds_from_casacore.py,sha256=gLZq0XAjeB4SDzE5nR1p3W-nmopshn_kA5l7kUbl9jo,42490
+xradio/image/_util/_casacore/xds_to_casacore.py,sha256=U21bCfJMm0SYEO-VyTnysjS2qnPY8yiHCnqKaBZ87Bw,15424
+xradio/image/_util/_fits/xds_from_fits.py,sha256=t1HefZZqKnG_G9a6soMiMbDXJA3Kt0aSStp1CnSuUrw,28461
 xradio/image/_util/_zarr/common.py,sha256=apMX_bF4Hr3pFGjnDFpp36KgmhTYAPBZquNkjBHrsXk,307
 xradio/image/_util/_zarr/xds_from_zarr.py,sha256=4b6KHmAcnrhBbCi-Z7e3Lm6l6wziJL1zaNIohmPAYDk,3601
 xradio/image/_util/_zarr/xds_to_zarr.py,sha256=wogXbwX8n3Sl9PHoc3_Y_LBowQsQ-94HZQFZ5NcxUZA,1624
@@ -26,9 +26,9 @@ xradio/image/_util/_zarr/zarr_low_level.py,sha256=fVaPnxYazz6UDLBZuyUOekZKQ945Ow
 xradio/measurement_set/__init__.py,sha256=0M0Zl5uKozIp5mgPhNaIoal_UuZc1HCA6_zQOhLFo8A,567
 xradio/measurement_set/convert_msv2_to_processing_set.py,sha256=FC54AApYh0WtQrK4Z5b7MiuRE0X99oziYEnDu4BOjMw,6591
 xradio/measurement_set/load_processing_set.py,sha256=RedC4i4dOHzgYiz_C0P-3F0PwvEGUomvq_ilpcfUCco,5540
-xradio/measurement_set/measurement_set_xds.py,sha256=WevpUY13ie43ji8gNZ73NAxExtnlCRJCEWhwgFIwyoE,4442
+xradio/measurement_set/measurement_set_xds.py,sha256=slZ4Nj3io3WkspjUqY4E-JO3LR2X7g9w2e88yjxcDQ8,4621
 xradio/measurement_set/open_processing_set.py,sha256=533fwEP4pJIBB-NdwFttbobVb8zEXsblvDTjF4ptTDU,3831
-xradio/measurement_set/processing_set.py,sha256=QisdrFKt-At9jGNDcpD2rOa5UonQ3npW8XzfMjCURAs,29898
+xradio/measurement_set/processing_set.py,sha256=uVc8PpG_7ivGYqTZpDCLCfuXBOwTLMFj3TebbEpKMRI,29904
 xradio/measurement_set/schema.py,sha256=OPAliKAdZt9oblyGW-CHnEzl-1hMj3-9I8G220XArXM,76439
 xradio/measurement_set/_utils/__init__.py,sha256=XE-h1yMfr6tVD6gdUwXO1CVq5SQ6kD_oj-e5TFwslds,97
 xradio/measurement_set/_utils/msv2.py,sha256=7hnZMFoQ-s1g0ATjEupLvtdqQCdroPv-Rl5OwjqXjh8,4430
@@ -69,8 +69,8 @@ xradio/schema/metamodel.py,sha256=WjtW7pAVzcjLRWifRH3sQoOiN6TV810hARpOIz1M_gw,38
 xradio/schema/typing.py,sha256=8-o6fZd99kJ4FVdgBYRTIRJ-wDqpcUNXzCTfJvl3TIw,10439
 xradio/sphinx/__init__.py,sha256=VGY-7Ty3q67qpnBee0-znbiJ-Iy0F93UO--IpjEdHXc,380
 xradio/sphinx/schema_table.py,sha256=YTQvK-VOBIpFtcx7sjcaMod4OSbl0uowelYIcdP3oVg,11878
-xradio-0.0.43.dist-info/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
-xradio-0.0.43.dist-info/METADATA,sha256=U887ZEkZ5FtkAADWgz1tSwoWfMBN1yVqsVxCa61GRfI,4488
-xradio-0.0.43.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-xradio-0.0.43.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
-xradio-0.0.43.dist-info/RECORD,,
+xradio-0.0.44.dist-info/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
+xradio-0.0.44.dist-info/METADATA,sha256=VAhya80uIfNOmDf2YCuAV-5px8TAaijN4zanDk1-_6A,4488
+xradio-0.0.44.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+xradio-0.0.44.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
+xradio-0.0.44.dist-info/RECORD,,

{xradio-0.0.43.dist-info → xradio-0.0.44.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.5.0)
 Root-Is-Purelib: true
 Tag: py3-none-any