xradio 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

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/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/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.dist-info → xradio-0.0.8.dist-info}/METADATA

@@ -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.dist-info → xradio-0.0.8.dist-info}/RECORD

@@ -1,7 +1,7 @@
 xradio/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xradio/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xradio/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/data/_dropbox.py,sha256=1O6685_YYtYV5mP9Zoyh1tq6U5K8Evj937rq4sBiLkw,2137
+xradio/data/_dropbox.py,sha256=PUz25GUfgjOzzV0-Aim1lF7LSwT3TocPdy2XHlrk7Jo,2288
 xradio/data/_google_drive.py,sha256=GA3XabwzvA9lBGcHteHYQJhW8pbAXGhMCQBiNeWPphY,4340
 xradio/data/datasets.py,sha256=rBsum83Qe9v_SIRsD0Ym9roz0dyVKKZrysRXezwkDho,457
 xradio/image/__init__.py,sha256=OXBgZQK9f_EgSwW2Uz18GOwsc_OeU1XCHw9IWOybsZk,291
@@ -21,8 +21,8 @@ xradio/image/_util/_zarr/xds_from_zarr.py,sha256=Es97yMB74SY2D_ecY8c_y6M4Vm7eqmv
 xradio/image/_util/_zarr/xds_to_zarr.py,sha256=Sh2RbvKHLJx3vOVzDcrurNE215YWbKP1NFGM7zuQFRA,1642
 xradio/image/_util/_zarr/zarr_low_level.py,sha256=KRWx_UO3FwC9CDPTUh7k4ChotVJr6nPIdjkl8raaA9E,5452
 xradio/vis/__init__.py,sha256=u5CAJpknYDspPF_50istwz70FHSNc3eL8J76ndwgYZo,111
-xradio/vis/_processing_set.py,sha256=dpZbhGDvq4EQFBp5-XgUFJ48MYF897_21Xw77UF0SmE,905
-xradio/vis/convert_msv2_to_processing_set.py,sha256=liKLEpLTJLuIXCDKrxaokegnIlmUyu8lkBbR4egiMJg,25250
+xradio/vis/_processing_set.py,sha256=vk2mm66bGWxcIoGEk2gYJRPAhYDudB-7tpYmf-2hZ10,979
+xradio/vis/convert_msv2_to_processing_set.py,sha256=9hfm876CPUZRB1OM-dLBUCiNNx1AlBXk4Kk_LR4K-j8,25526
 xradio/vis/load_processing_set.py,sha256=HXQEjfZThK85sVhwp1JxxUTl1LqedZgAXQFyf9i5Wuw,4505
 xradio/vis/read_processing_set.py,sha256=E4nIWGz3fDkoOaF1uG09OcmstEN11Do3ekGIHONn2ac,688
 xradio/vis/vis_io.py,sha256=5b0htEaaAJ1XFJcR2HSNgQmxmmJNW-znrku7-QuPdSo,6022
@@ -50,8 +50,8 @@ xradio/vis/_vis_utils/_utils/stokes_types.py,sha256=rMX7BQFbGFfT6oP54qTit4q16JRA
 xradio/vis/_vis_utils/_utils/xds_helper.py,sha256=xrS46H7INWTxiuo2zN_a2wOaTr6irbVlkv1efjm7YxY,11853
 xradio/vis/_vis_utils/_zarr/read.py,sha256=fxznKzP0G8YFjkyMxg04PMI3Mtnu1mMjSnvEhY2arIU,7834
 xradio/vis/_vis_utils/_zarr/write.py,sha256=fnu9xh_AZ4OfoOGHsMnCpVEZ_2UiZGP7qXhlqZesCCs,9585
-xradio-0.0.7.dist-info/LICENSE.txt,sha256=dvACd-5O67yjSZlnEKcWmu3DqwzBtbC922iPv0KOeAw,1516
-xradio-0.0.7.dist-info/METADATA,sha256=F28JH2wv3UHZHUItldi_W3M3HMP_Aw7YZiPhHfXzMC0,4000
-xradio-0.0.7.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-xradio-0.0.7.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
-xradio-0.0.7.dist-info/RECORD,,
+xradio-0.0.8.dist-info/LICENSE.txt,sha256=dvACd-5O67yjSZlnEKcWmu3DqwzBtbC922iPv0KOeAw,1516
+xradio-0.0.8.dist-info/METADATA,sha256=eqF7XXFxnCMosiqCcXr4rOTl21madMMW19H925MLWKU,4000
+xradio-0.0.8.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+xradio-0.0.8.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
+xradio-0.0.8.dist-info/RECORD,,