xradio 0.0.7__tar.gz → 0.0.8__tar.gz

{xradio-0.0.7/src/xradio.egg-info → xradio-0.0.8}/PKG-INFO

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

{xradio-0.0.7 → xradio-0.0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xradio"
-version = "0.0.7"
+version = "0.0.8"
 description = "Xarray Radio Astronomy Data IO"
 authors = [
     {name = "Jan-Willem Steeb", email="jsteeb@nrao.edu"},

{xradio-0.0.7 → xradio-0.0.8}/src/xradio/data/_dropbox.py

@@ -8,6 +8,13 @@ from tqdm import tqdm
 #https://www.dropbox.com/scl/fi/x8tp0wu21gssbd1gxrnmy/Antennae_North.cal.lsrk.vis.zarr.zip?rlkey=l9jdr6tvyq4pe3381gukuly0d&dl=0
 
 FILE_ID = {
+ 'AA2-Mid-sim_00000.ms':
+  {
+    'file':'AA2-Mid-sim_00000.ms.zip',
+    'id':'2buf75ebhurjlfhe123qs',
+    'rlkey':'8wzgiavzxfp4aza4sx84qanid&dl'
+  },
+
   'Antennae_M8.img.zarr':
   {
     'file':'Antennae_M8.img.zarr.zip',

{xradio-0.0.7 → xradio-0.0.8}/src/xradio/vis/_processing_set.py

@@ -17,3 +17,6 @@ class _processing_set(dict):
             summary_data["end_frequency"].append(value["frequency"].values[-1])
         summary_df = pd.DataFrame(summary_data)
         return summary_df
+        
+    def get(self,id):
+        return self[list(self.keys())[id]]

{xradio-0.0.7 → xradio-0.0.8}/src/xradio/vis/convert_msv2_to_processing_set.py

@@ -633,14 +633,16 @@ def get_unqiue_intents(infile):
         rename_ids=subt_rename_ids["STATE"],
     )
 
-    obs_mode_dict = {}
-    for i, obs_mode in enumerate(state_xds.obs_mode.values):
-        if obs_mode in obs_mode_dict:
-            obs_mode_dict[obs_mode].append(i)
-        else:
-            obs_mode_dict[obs_mode] = [i]
-
-    return list(obs_mode_dict.keys()), obs_mode_dict.values()
+    if len(state_xds.data_vars) > 0:
+        obs_mode_dict = {}
+        for i, obs_mode in enumerate(state_xds.obs_mode.values):
+            if obs_mode in obs_mode_dict:
+                obs_mode_dict[obs_mode].append(i)
+            else:
+                obs_mode_dict[obs_mode] = [i]
+        return list(obs_mode_dict.keys()), list(obs_mode_dict.values())
+    else:  # empty state table
+        return ["None"], [0]
 
 
 def enumerated_product(*args):
@@ -674,8 +676,13 @@ def convert_msv2_to_processing_set(
         field_ids = np.arange(read_generic_table(infile, "FIELD").dims["row"])
     elif partition_scheme == "ddi_state":
         state_xds = read_generic_table(infile, "STATE")
-        state_ids = np.arange(state_xds.dims["row"])
-        intents = state_xds.obs_mode.values
+
+        if len(state_xds.data_vars) > 0:
+            state_ids = [np.arange(state_xds.dims["row"])]
+            intents = state_xds.obs_mode.values
+        else:  # empty state table
+            state_ids = [0]
+            intents = ["None"]
         # print(state_xds, intents)
         # field_ids = [None]
         field_ids = np.arange(read_generic_table(infile, "FIELD").dims["row"])

{xradio-0.0.7 → xradio-0.0.8/src/xradio.egg-info}/PKG-INFO

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