xarray-ms 0.3.9__tar.gz → 0.5.0__tar.gz

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xarray-ms
-Version: 0.3.9
+Version: 0.5.0
 Summary: xarray MSv4 views over MSv2 Measurement Sets
 Author-email: Simon Perkins <simon.perkins@gmail.com>
 License-File: LICENSE

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/doc/source/changelog.rst

@@ -3,6 +3,14 @@
 Changelog
 =========
 
+0.5.0 (02-03-2026)
+------------------
+* Fix ``VisibilityCoder`` typo (:pr:`145`)
+* Raise an exception if no MSv4 Frame Type is available in ``EpochCoder.decode`` (:pr:`145`)
+* Replace an assert with an ``InvalidMeasurementSet`` exception if the
+  number of ``TabRefTypes`` doesn't match the ``TabRefCodes`` (:pr:`145`)
+* Guard frame inference in a dictionary with a KeyError (:pr:`145`)
+
 0.3.9 (18-02-2026)
 ------------------
 * Define and use an ``OnMissingLiteralType`` (:pr:`144`)

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/doc/source/conf.py

@@ -11,7 +11,7 @@
 project = "xarray-ms"
 copyright = "2024 - 2025 NRF (SARAO) and Rhodes University (RATT) Centre"
 author = "Simon Perkins"
-release = "0.3.9"
+release = "0.5.0"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xarray-ms"
-version = "0.3.9"
+version = "0.5.0"
 description = "xarray MSv4 views over MSv2 Measurement Sets"
 authors = [{name = "Simon Perkins", email = "simon.perkins@gmail.com"}]
 readme = "README.rst"
@@ -54,7 +54,7 @@ extend-select = ["I"]
 # github_url = "https://github.com/<user or organization>/<project>/"
 
 [tool.tbump.version]
-current = "0.3.9"
+current = "0.5.0"
 
 # Example of a semver regexp.
 # Make sure this matches current_version before

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/xarray_ms/backend/msv2/factories/correlated.py

@@ -23,7 +23,7 @@ from xarray_ms.backend.msv2.imputation import (
 from xarray_ms.backend.msv2.measures_encoders import (
   MSv2Coder,
   MSv2CoderFactory,
-  VisiblityCoder,
+  VisibilityCoder,
 )
 from xarray_ms.backend.msv2.structure import MSv2StructureFactory, PartitionKeyT
 from xarray_ms.casa_types import ColumnDesc, Polarisations
@@ -57,7 +57,7 @@ MSV4_to_MSV2_COLUMN_SCHEMAS = {
   ),
   "FLAG": MSv2ColumnSchema("FLAG", ("frequency", "polarization"), 1, None),
   "VISIBILITY": MSv2ColumnSchema(
-    "DATA", ("frequency", "polarization"), np.nan + np.nan * 1j, VisiblityCoder()
+    "DATA", ("frequency", "polarization"), np.nan + np.nan * 1j, VisibilityCoder()
   ),
   "WEIGHT_ROW": MSv2ColumnSchema(
     "WEIGHT",

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/xarray_ms/backend/msv2/measures_adapters.py

@@ -380,7 +380,11 @@ class ArrowTableMeasuresAdapter(ColumnDescMeasuresAdapter, MeasuresMixin):
       # The measures information may contain it's own reference types and codes
       if isinstance(tab_ref_types := measinfo.get("TabRefTypes"), Sequence):
         if isinstance(tab_ref_codes := measinfo.get("TabRefCodes"), Sequence):
-          assert len(tab_ref_types) == len(tab_ref_codes)
+          if len(tab_ref_types) != len(tab_ref_codes):
+            raise InvalidMeasurementSet(
+              f"TabRefTypes {tab_ref_types} and TabRefCodes {tab_ref_codes} "
+              f"have different lengths in {self.column_name} MEASINFO"
+            )
         else:
           tab_ref_codes = [MeasuresEnum[t].value for t in tab_ref_types]
 
@@ -390,7 +394,15 @@ class ArrowTableMeasuresAdapter(ColumnDescMeasuresAdapter, MeasuresMixin):
         code_frame_map = {m.value: m.name for m in MeasuresEnum}
 
       # Try and find a unique frame
-      if len(frames := {code_frame_map[c] for c in var_ref_col}) == 0:
+      try:
+        frames = {code_frame_map[c] for c in var_ref_col}
+      except KeyError as e:
+        raise InvalidMeasurementSet(
+          f"Code {e.args[0]} in {var_ref_col_name} is not in the "
+          f"code frame map {code_frame_map} for column {self.column_name}"
+        ) from e
+
+      if len(frames) == 0:
         return None
 
       if len(frames) > 1:

{xarray_ms-0.3.9 → xarray_ms-0.5.0}/xarray_ms/backend/msv2/measures_encoders.py

@@ -107,21 +107,6 @@ class MeasuresCoder(MSv2Coder):
     self.measures_adapter = measures_adapter
 
 
-class QuantityCoder(MeasuresCoder):
-  """Encodes a quantum unit"""
-
-  def encode(self, variable: Variable, name: T_Name = None) -> Variable:
-    dims, data, attrs, encoding = unpack_for_encoding(variable)
-    attrs = {k: v for k, v in attrs.items() if k not in {"type", "units"}}
-    return Variable(dims, data, attrs, encoding, fastpath=True)
-
-  def decode(self, variable: Variable, name: T_Name = None) -> Variable:
-    dims, data, attrs, encoding = unpack_for_decoding(variable)
-    attrs["type"] = "quantity"
-    attrs["units"] = self.measures_adapter.quantum_unit("raise")
-    return Variable(dims, data, attrs, encoding, fastpath=True)
-
-
 class SuppliedAttributesCoder(MSv2Coder):
   """Adds and removes the supplied attributes during decoding and encoding"""
 
@@ -149,7 +134,7 @@ class SuppliedQuantityCoder(SuppliedAttributesCoder):
     super().__init__({"type": "quantity", "units": units})
 
 
-class VisiblityCoder(SuppliedQuantityCoder):
+class VisibilityCoder(SuppliedQuantityCoder):
   """Visibility coder"""
 
   def __init__(self):
@@ -227,7 +212,7 @@ class EpochCoder(MeasuresCoder):
 
   def decode(self, variable: Variable, name: T_Name = None) -> Variable:
     dims, data, attrs, encoding = unpack_for_decoding(variable)
-    attrs["type"] = self.measures_adapter.msv4_type()
+    attrs["type"] = self.measures_adapter.msv4_type("raise")
     attrs["units"] = self.measures_adapter.quantum_unit("raise")
     attrs["format"] = "unix"
     if (msv2_frame := self.measures_adapter.msv2_frame("raise")) == "UTC":