xradio 1.1.0__py3-none-any.whl → 1.1.12a0__py3-none-any.whl

xradio/_utils/zarr/common.py

@@ -1,6 +1,7 @@
 import xarray as xr
 import os
 
+
 # from xradio.vis._vis_utils._ms.msv2_to_msv4_meta import (
 #     column_description_casacore_to_msv4_measure,
 # )

xradio/image/_util/_casacore/xds_from_casacore.py

@@ -381,12 +381,14 @@ def _casa_image_to_xds_coords(
     attrs["sphr_dims"] = sphr_dims
     coords = {}
     coord_attrs = {}
-    coords["time"], coord_attrs["time"] = _get_time_values_attrs(coord_dict)
-    coords["frequency"], coord_attrs["frequency"] = _get_freq_values_attrs(csys, shape)
-    velocity_vals, coord_attrs["velocity"] = _get_velocity_values_attrs(
+    (coords["time"], coord_attrs["time"]) = _get_time_values_attrs(coord_dict)
+    (coords["frequency"], coord_attrs["frequency"]) = _get_freq_values_attrs(
+        csys, shape
+    )
+    (velocity_vals, coord_attrs["velocity"]) = _get_velocity_values_attrs(
         coord_dict, coords["frequency"]
     )
-    coords["polarization"], coord_attrs["polarization"] = _get_pol_values_attrs(
+    (coords["polarization"], coord_attrs["polarization"]) = _get_pol_values_attrs(
         coord_dict
     )
     coords["velocity"] = (["frequency"], velocity_vals)
@@ -683,7 +685,6 @@ def _get_persistent_block(
     block = _read_image_chunk(infile, shapes, starts)
     block = np.expand_dims(block, new_axes)
     block = block.transpose(transpose_list)
-    block = da.from_array(block, chunks=block.shape)
     block = xr.DataArray(block, dims=dimorder)
     return block
 

xradio/image/_util/_zarr/common.py

@@ -1,5 +1,6 @@
 import numpy as np
 
+
 _np_types = {
     "complex128": np.complex128,
     "complex64": np.complex64,

xradio/image/_util/casacore.py

@@ -202,6 +202,7 @@ def _xds_to_multiple_casa_images(xds: xr.Dataset, image_store_name: str) -> None
                             ]
 
                         if "flag" in xds.attrs["data_groups"][data_group]:
+                            print("******* found flag")
                             mask_sky_name = xds.attrs["data_groups"][data_group]["flag"]
                             image_to_write_xds["MASK_0"] = xds[mask_sky_name]
                             image_to_write_xds["SKY"].attrs["flag"] = "MASK_0"

xradio/image/_util/image_factory.py

@@ -337,16 +337,7 @@ def detect_store_type(store):
                 raise ValueError("Unknown directory structure." + str(store))
         else:
             logger.error("Path does not exist.")
-            raise ValueError(
-                "Path does not exist. The current path: "
-                + str(os.system("pwd"))
-                + " .The current casa directory: "
-                + str(os.system("ls 3c286_Band6_5chans_lsrk_robust_0.5_niter_99_casa"))
-                + ". The current fits directory: "
-                + str(os.system("ls 3c286_Band6_5chans_lsrk_robust_0.5_niter_99_fits"))
-                + " The given store "
-                + str(store)
-            )
+            raise ValueError("Path does not exist." + str(store))
     else:
         store_type = "zarr"
 

xradio/measurement_set/_utils/_msv2/_tables/read.py

@@ -93,12 +93,9 @@ def convert_mjd_time(rawtimes: np.ndarray) -> np.ndarray:
     np.ndarray
         times converted to pandas reference and datetime type
     """
-
-    print("^^^^^^^", rawtimes, MJD_DIF_UNIX, SECS_IN_DAY)
     times_reref = pd.to_datetime(
         (rawtimes - MJD_DIF_UNIX) * SECS_IN_DAY, unit="s"
     ).values
-    print("^^^^^^^", times_reref)
 
     return times_reref
 
@@ -163,7 +160,7 @@ def make_taql_where_between_min_max(
     if min_max_range is None:
         taql = None
     else:
-        min_val, max_val = min_max_range
+        (min_val, max_val) = min_max_range
         taql = f"where {colname} >= {min_val} AND {colname} <= {max_val}"
 
     return taql
@@ -229,7 +226,7 @@ def find_projected_min_max_array(
     """Does the min/max checks and search for find_projected_min_max_table()"""
 
     sorted_array = np.sort(array)
-    range_min, range_max = min_max
+    (range_min, range_max) = min_max
     if len(sorted_array) < 2:
         tol = np.finfo(sorted_array.dtype).eps * 4
     else:
@@ -894,8 +891,7 @@ def raw_col_data_to_coords_vars(
 
     if col in timecols:
         if col == "MJD":
-            # data = convert_mjd_time(data).astype("float64") / 1e9
-            data = convert_mjd_time(data).astype("datetime64[ns]").view("int64") / 1e9
+            data = convert_mjd_time(data).astype("float64") / 1e9
         else:
             try:
                 data = convert_casacore_time(data, False)

xradio/measurement_set/_utils/_msv2/create_field_and_source_xds.py

@@ -285,7 +285,6 @@ def extract_ephemeris_info(
     # Metadata has to be fixed manually. Alternatively, issues like
     # UNIT/QuantumUnits issue could be handled in convert_generic_xds_to_xradio_schema,
     # but for now preferring not to pollute that function.
-
     time_ephemeris_dim = ["time_ephemeris"]
     to_new_data_variables = {
         # mandatory: SOURCE_RADIAL_VELOCITY

xradio/measurement_set/_utils/_msv2/msv4_sub_xdss.py

@@ -27,6 +27,7 @@ from xradio.measurement_set._utils._msv2._tables.read import (
     table_has_column,
 )
 
+
 standard_time_coord_attrs = make_time_measure_attrs(time_format="unix")
 
 
@@ -80,7 +81,6 @@ def rename_and_interpolate_to_time(
     )
 
     # rename the time_* axis to time.
-
     time_coord = {"time": (time_initial_name, interp_time.data)}
     renamed_time_xds = interpolated_xds.assign_coords(time_coord)
     renamed_time_xds.coords["time"].attrs.update(standard_time_coord_attrs)

xradio/measurement_set/_utils/_utils/interpolate.py

@@ -41,10 +41,6 @@ def interpolate_to_time(
             method = "linear"
         else:
             method = "nearest"
-
-        # print("xds before interp:",xds.NORTH_POLE_ANGULAR_DISTANCE.values, xds[time_name].values)
-        # print("interp_time data:",interp_time,interp_time.data)
-        # print("method:",method)
         xds = xds.interp(
             {time_name: interp_time.data}, method=method, assume_sorted=True
         )
@@ -60,6 +56,5 @@ def interpolate_to_time(
             f"{message_prefix}: interpolating the time coordinate "
             f"from {points_before} to {points_after} points"
         )
-        # print("xds after interp:",xds.NORTH_POLE_ANGULAR_DISTANCE.values, xds[time_name].values)
 
     return xds

xradio/measurement_set/_utils/_utils/partition_attrs.py

@@ -2,6 +2,7 @@ from typing import Dict, TypedDict, Union
 
 import xarray as xr
 
+
 PartitionIds = TypedDict(
     "PartitionIds",
     {

xradio/schema/common.py

@@ -9,6 +9,7 @@ from xradio.schema.bases import (
 from xradio.schema.typing import Attr, Coord, Coordof, Data, Dataof, Name
 import numpy
 
+
 # Dimensions
 Time = Literal["time"]
 """ Observation time dimension """

xradio/testing/measurement_set/msv2_io.py

@@ -19,6 +19,7 @@ from casacore.tables import default_ms, default_ms_subtable
 from casacore.tables.tableutil import makedminfo, maketabdesc
 from casacore.tables.msutil import complete_ms_desc, makearrcoldesc, required_ms_desc
 
+
 # 2 observations, 2 fields, 2 states
 # 2 SPWs, 4 polarizations
 default_ms_descr = {

{xradio-1.1.0.dist-info → xradio-1.1.12a0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xradio
-Version: 1.1.0
+Version: 1.1.12a0
 Summary:  Xarray Radio Astronomy Data IO
 Author-email: Jan-Willem Steeb <jsteeb@nrao.edu>, Federico Montesino Pouzols <pouzols@eso.edu>, Dave Mehringer <dmehring@nrao.edu>, Peter Wortmann <peter.wortmann@skao.int>
 License: BSD 3-Clause License

{xradio-1.1.0.dist-info → xradio-1.1.12a0.dist-info}/RECORD

@@ -8,22 +8,22 @@ xradio/_utils/xarray_helpers.py,sha256=LQyh754BRFvKz8xP0CF_nWBj0s2HI5KVUz7c1YxC3
 xradio/_utils/_casacore/casacore_from_casatools.py,sha256=lpkiDVj6eXpxY07xwTOwZba_Mw-g2uOAdspOdfPVVtE,31208
 xradio/_utils/_casacore/tables.py,sha256=puRidbjtVx6caEG_Z5TebTLdTUbtBUhzvqByKLQTHfo,1389
 xradio/_utils/zarr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/_utils/zarr/common.py,sha256=MTew96RhSS2f77-hIsP4aXS2tZwF1THvjpkRQQqpmgk,2989
+xradio/_utils/zarr/common.py,sha256=Gy4Nl4JrISzLlBq8QqLwuWYUS7eDsgT3q9s2dhrNiNc,2990
 xradio/image/__init__.py,sha256=s18tPMOv0VLW7zrq9cQ_gOb577YXfqfK8ZMhA63LGH0,468
 xradio/image/image.py,sha256=SuEc0G4MRIOAv--qSPo2jMfJhJlHGA-nr3LdSA_Sykk,15366
 xradio/image/image_xds.py,sha256=-RS3wM6uSspZSu50jaPgopUYQQbU8KgQSBUt6sh7XuE,9169
 xradio/image/schema.py,sha256=S3FkclHMLucgw3ciTpqdc9x1P--ZxwhJZis5lyW7XcM,6043
 xradio/image/_util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/image/_util/casacore.py,sha256=eyV9Zb7ubMb8YD0xkrRCHa1C7btBF7w5SBrXzBFQZ1E,10028
+xradio/image/_util/casacore.py,sha256=TpMdAl69ag8fGqYMaa9iJ98bEx_gXg-y3NdV_d_42hI,10084
 xradio/image/_util/common.py,sha256=lcxletfv18zUUJgG8Qs_fV-VdON3D3xJIo5r97be6FU,8490
-xradio/image/_util/image_factory.py,sha256=qlQRfOBF_i98embr4x7vComrKkheRRlHl4J2ETTsAVo,24905
+xradio/image/_util/image_factory.py,sha256=51hUnwBSQaMK5KucW8tlNkbvDVwJaVNgV97T_KneFug,24484
 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=PHt7SNllwYW70w9pEhLrcCZ7avoZNmwKxTSnhSQQtOU,1793
-xradio/image/_util/_casacore/xds_from_casacore.py,sha256=gdyMQ7LymKLU0Y1rj6uypGzTOkJZ5GWS-uCi-N6u4es,45913
+xradio/image/_util/_casacore/xds_from_casacore.py,sha256=0gAxdwINDCoFsTdSQIRYu8MCxm-oc-mLtv7YJADybN8,45882
 xradio/image/_util/_casacore/xds_to_casacore.py,sha256=dltoHo9TRklbxjcSkiGu6KaSehNQarGqo3lBSCiOhAw,20331
 xradio/image/_util/_fits/xds_from_fits.py,sha256=U5bskcCCsQAUn8spJ6fxXtgGIiGaNflK4yS26x6KH9o,38221
-xradio/image/_util/_zarr/common.py,sha256=U_nMtFYN_qFDP0h8JiLEEBG3ufF5ckVo8XQMC16xti8,307
+xradio/image/_util/_zarr/common.py,sha256=ltlj3uFa-uv8lXlDtV79QnfNmfm0tyhXN5FDAjZtjzg,308
 xradio/image/_util/_zarr/xds_from_zarr.py,sha256=KMsfaSSm9kyVoztS6pUzGNxMZzQnCxkk0kDv2GxW5Kw,4451
 xradio/image/_util/_zarr/xds_to_zarr.py,sha256=nsDvDD-kuMuMF2dDlj0jTxSW4mdR-jjIsvXHi5uIERU,2373
 xradio/image/_util/_zarr/zarr_low_level.py,sha256=xnYm6EmVbmLxMlOSXH32SABfQBLHfr2H9ch9gYwFNXs,13338
@@ -38,24 +38,24 @@ xradio/measurement_set/_utils/__init__.py,sha256=iTbgPiQPw9hyXSdQYPcwGRrP2PFXgF6
 xradio/measurement_set/_utils/_msv2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xradio/measurement_set/_utils/_msv2/conversion.py,sha256=2ASFGgF2L-JiYzFTQhUd3QvTdqPRjd1I_oy87AfQf5g,54836
 xradio/measurement_set/_utils/_msv2/create_antenna_xds.py,sha256=s3HnTbyOl7h6pAocJ4dpd095s7qHSSngeQ34Ygo8Uk0,17650
-xradio/measurement_set/_utils/_msv2/create_field_and_source_xds.py,sha256=unKWb-bIjZ9GoN-jbUzTAZ-GGf7HU4DWLl9km-xzGho,33558
+xradio/measurement_set/_utils/_msv2/create_field_and_source_xds.py,sha256=k5pJhIafu9QANaX9-pdJ1fOqGOtplHxR2Ve9xBCf3dA,33557
 xradio/measurement_set/_utils/_msv2/msv2_to_msv4_meta.py,sha256=ZpxJJ2yTjjEy6A4XWS9eT-1J92jLwR8CNyoDBfhAVm8,1572
 xradio/measurement_set/_utils/_msv2/msv4_info_dicts.py,sha256=cMgD6EB1pRShL6TBVSpom0VhkqTnsYNi9vxz_gfgGtU,11952
-xradio/measurement_set/_utils/_msv2/msv4_sub_xdss.py,sha256=nXuiDhszP3AejUo1zlpqBcyL-R5bsRw-KTOP3M2-HI0,30453
+xradio/measurement_set/_utils/_msv2/msv4_sub_xdss.py,sha256=6BXZkypfe1C4n5bQDtxMdY0qnVuO3DDjtwDLUwtUOow,30453
 xradio/measurement_set/_utils/_msv2/optimised_functions.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xradio/measurement_set/_utils/_msv2/partition_queries.py,sha256=_eWxxgxEYRX1Pj1ZFHD_HIcrF9Rpfe6OJmp3uYJ2UiY,11970
 xradio/measurement_set/_utils/_msv2/subtables.py,sha256=mrf7g7mbC4crtnQ0wFocPcFRNlvq_7e7iDzE5B6ugl4,843
-xradio/measurement_set/_utils/_msv2/_tables/read.py,sha256=P5-sz_hwBRqzgPfhN-zcdFayfOekDMQJKLN2f8OTnys,45599
+xradio/measurement_set/_utils/_msv2/_tables/read.py,sha256=tW0ROevR4tXpydIWuj2UyNhWNtvkKU9NkYxMi1ZoVKc,45421
 xradio/measurement_set/_utils/_msv2/_tables/read_main_table.py,sha256=FFoTVpY8JNvcXnTvUiraWNU9gB9-_9GR7he0NbODabg,3168
 xradio/measurement_set/_utils/_msv2/_tables/table_query.py,sha256=Uc1zeiU-rYtCsYXpij8jzFG5NPBYvIh70qW1srn7B98,1498
-xradio/measurement_set/_utils/_utils/interpolate.py,sha256=eIHEZ2VAW0JjMQQ08xaUcKQpko7HrQU2UfXKzVBlNwA,2262
-xradio/measurement_set/_utils/_utils/partition_attrs.py,sha256=Uxtia6HPfPUcWfsT1cJAVTqoOS0zedZEsqgI9NWx1z4,3400
+xradio/measurement_set/_utils/_utils/interpolate.py,sha256=LlDYi-h7aj_tsuZqAu5Kr-vOi6fVBPpVbew-i-HnmA0,1962
+xradio/measurement_set/_utils/_utils/partition_attrs.py,sha256=JaePHts_A0EbB4K-0a_uC98RZ2EmfjB9pDSEI11oAwk,3401
 xradio/measurement_set/_utils/_utils/stokes_types.py,sha256=DMa8TmmS7BQ99Xm8c7ZjcRapMtLbrKVxrt4f0qUIOvg,561
 xradio/measurement_set/_utils/_zarr/encoding.py,sha256=ze5ncHEBa-sVwJCayJa2irOdaxZEkYfBKObuij7uVRc,371
 xradio/schema/__init__.py,sha256=tJqTUSwkwvkak_6eAZqhT5BQqunlfxciG07-y8uSrMY,683
 xradio/schema/bases.py,sha256=TRwvOo2UA7ki-yV1Vcb5aDxAAxH2lTKtwoGx2m5VB4o,17512
 xradio/schema/check.py,sha256=6bndBpmzkdiM3Ca1lquYEDb65H2ZZOGp2xQyIjqNHqU,23804
-xradio/schema/common.py,sha256=Xv_pbTWfgl_K6GT3l2yYi0aKvp8X-9DSFT-bIkX7i20,2015
+xradio/schema/common.py,sha256=Hi2vT8s0za0J7FfA0OhzrGdtwjJ5eQWH7x3hfhBCOuo,2016
 xradio/schema/dataclass.py,sha256=gzNqXzkBLV7ETEyddNOd_jRgP8rkivCkfUlURLz8z5c,19300
 xradio/schema/export.py,sha256=qN8O8Wc9FtrYbiEmg1u66cN4coY_8QVHrZfb-oHhYp0,3510
 xradio/schema/metamodel.py,sha256=F0lxnn0wLrsT2HrBo2zfIcmRLxD2yQwh7XfDciPNwVs,4963
@@ -67,9 +67,9 @@ xradio/testing/_utils/__template__.py,sha256=1A3WJ2dcQwiIsxHqJscvn1Pmff7FhUXXxbp
 xradio/testing/measurement_set/__init__.py,sha256=Vjk_959ZT48O7JRwXnUrIC3YvTH8GnrNRbee-RLjDco,1618
 xradio/testing/measurement_set/checker.py,sha256=39wbX0xcLm6gsPiVTASbOciuoF_WzGOAcVe-7wir0Xg,4566
 xradio/testing/measurement_set/io.py,sha256=A3m6tm2Rvoacw8-SO3UhNboEek1YeDCMguwHClUIqn4,559
-xradio/testing/measurement_set/msv2_io.py,sha256=twQofNpkc8cIU2Z4CGpFL1fV_JotUWDX40nmGFVrfH8,59713
-xradio-1.1.0.dist-info/licenses/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
-xradio-1.1.0.dist-info/METADATA,sha256=Mcoe3VygnEKSktsS4L3juXKZ-o2TaMtV7nojyi4RNL0,9744
-xradio-1.1.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-xradio-1.1.0.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
-xradio-1.1.0.dist-info/RECORD,,
+xradio/testing/measurement_set/msv2_io.py,sha256=D6ffVOwcFcWtHJWzxOPg5J1zWh_rKsbPsgYjvdXjOos,59714
+xradio-1.1.12a0.dist-info/licenses/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
+xradio-1.1.12a0.dist-info/METADATA,sha256=HXt_FY9cTsB8bzaeTbkPhBS-M265fI2dq-QuM5jpiCE,9747
+xradio-1.1.12a0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xradio-1.1.12a0.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
+xradio-1.1.12a0.dist-info/RECORD,,

{xradio-1.1.0.dist-info → xradio-1.1.12a0.dist-info}/WHEEL

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