xarray-ms 0.2.8__tar.gz → 0.2.9__tar.gz

{xarray_ms-0.2.8 → xarray_ms-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xarray-ms
-Version: 0.2.8
+Version: 0.2.9
 Summary: xarray MSv4 views over MSv2 Measurement Sets
 Author: Simon Perkins
 Author-email: simon.perkins@gmail.com

{xarray_ms-0.2.8 → xarray_ms-0.2.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "xarray-ms"
-version = "0.2.8"
+version = "0.2.9"
 description = "xarray MSv4 views over MSv2 Measurement Sets"
 authors = ["Simon Perkins <simon.perkins@gmail.com>"]
 readme = "README.rst"
@@ -58,7 +58,7 @@ build-backend = "poetry.core.masonry.api"
 # github_url = "https://github.com/<user or organization>/<project>/"
 
 [tool.tbump.version]
-current = "0.2.8"
+current = "0.2.9"
 
 # Example of a semver regexp.
 # Make sure this matches current_version before

{xarray_ms-0.2.8 → xarray_ms-0.2.9}/xarray_ms/backend/msv2/imputation.py

@@ -17,6 +17,12 @@ def _maybe_return_table_or_max_id(
 ) -> pa.Table | int:
   """Returns the existing table if a row entry exists,
   else returns the maximum id"""
+
+  # NOTE(sjperkins)
+  # Negative ids are invalid foreign keys
+  # AFAICT this implies there's no entry in the associated subtable
+  # Set a ceiling of 0 for imputation purposes
+  ids = np.concatenate([ids, [0]])
   max_id = np.max(ids)
 
   if max_id < len(table):

{xarray_ms-0.2.8 → xarray_ms-0.2.9}/xarray_ms/testing/simulator.py

@@ -413,6 +413,7 @@ class MSStructureSimulator:
     feed = desc.feed_map[tuple(corrs.polarisation_types())]
 
     array_groups: List[Tuple[List[str], List[npt.NDArray]]] = [
+      (["PROCESSOR_ID"], [desc.PROCESSOR_ID]),
       (["OBSERVATION_ID"], [desc.OBSERVATION_ID]),
       (["FIELD_ID"], [desc.FIELD_ID]),
       (["DATA_DESC_ID"], [desc.DATA_DESC_ID]),