imap-processing 0.12.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

imap_processing/ultra/utils/ultra_l1_utils.py

@@ -6,7 +6,9 @@ from imap_processing.cdf.imap_cdf_manager import ImapCdfAttributes
 
 
 def create_dataset(
-    data_dict: dict, name: str, level: str, data_version: str
+    data_dict: dict,
+    name: str,
+    level: str,
 ) -> xr.Dataset:
     """
     Create xarray for L1b data.
@@ -19,8 +21,6 @@ def create_dataset(
         Name of the dataset.
     level : str
         Level of the dataset.
-    data_version : str
-        Version of the data.
 
     Returns
     -------
@@ -30,7 +30,6 @@ def create_dataset(
     cdf_manager = ImapCdfAttributes()
     cdf_manager.add_instrument_global_attrs("ultra")
     cdf_manager.add_instrument_variable_attrs("ultra", level)
-    cdf_manager.add_global_attribute("Data_version", data_version)
 
     # L1b extended spin, badtimes, and cullingmask data products
     if "spin_number" in data_dict.keys():
@@ -38,7 +37,7 @@ def create_dataset(
             "spin_number": data_dict["spin_number"],
             "energy_bin_geometric_mean": data_dict["energy_bin_geometric_mean"],
             # Start time aligns with the universal spin table
-            "epoch": data_dict["spin_start_time"],
+            "epoch": data_dict["epoch"],
         }
         default_dimension = "spin_number"
     # L1c pset data products
@@ -79,6 +78,8 @@ def create_dataset(
         "velocity_sc",
         "velocity_dps_sc",
         "velocity_dps_helio",
+        "direct_event_unit_velocity",
+        "direct_event_unit_position",
     }
     rates_keys = {
         "ena_rates",
@@ -86,37 +87,37 @@ def create_dataset(
         "quality_ena_rates",
     }
 
-    for key in data_dict.keys():
+    for key, data in data_dict.items():
         # Skip keys that are coordinates.
         if key in ["epoch", "spin_number", "energy_bin_geometric_mean", "healpix"]:
             continue
         elif key in velocity_keys:
             dataset[key] = xr.DataArray(
-                data_dict[key],
+                data,
                 dims=["epoch", "component"],
                 attrs=cdf_manager.get_variable_attributes(key),
             )
-        elif key == "ena_rates_threshold":
+        elif key in ("ena_rates_threshold", "energy_bin_delta"):
             dataset[key] = xr.DataArray(
-                data_dict[key],
+                data,
                 dims=["energy_bin_geometric_mean"],
                 attrs=cdf_manager.get_variable_attributes(key),
             )
         elif key in rates_keys:
             dataset[key] = xr.DataArray(
-                data_dict[key],
+                data,
                 dims=["energy_bin_geometric_mean", "spin_number"],
                 attrs=cdf_manager.get_variable_attributes(key),
             )
         elif key == "counts":
             dataset[key] = xr.DataArray(
-                data_dict[key],
+                data,
                 dims=["energy_bin_geometric_mean", "healpix"],
                 attrs=cdf_manager.get_variable_attributes(key),
             )
         else:
             dataset[key] = xr.DataArray(
-                data_dict[key],
+                data,
                 dims=[default_dimension],
                 attrs=cdf_manager.get_variable_attributes(key),
             )

imap_processing/utils.py

@@ -352,7 +352,7 @@ def packet_file_to_datasets(
         # to the packet_name being a part of the variable name in the XTCE definition
         ds = ds.rename(
             {
-                # partition splits the string into 3 parts: before ".", ".", after "."
+                # partition splits the string into 3 parts: before ".", after "."
                 # if there was no ".", the second part is an empty string, so we use
                 # the original key in that case
                 key: key.partition(".")[2] or key

{imap_processing-0.12.0.dist-info → imap_processing-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imap-processing
-Version: 0.12.0
+Version: 0.13.0
 Summary: IMAP Science Operations Center Processing
 License: MIT
 Keywords: IMAP,SDC,SOC,Science Operations
@@ -30,7 +30,7 @@ Provides-Extra: tools
 Requires-Dist: astropy-healpix (>=1.0)
 Requires-Dist: cdflib (>=1.3.1,<2.0.0)
 Requires-Dist: healpy (>=1.18.0,<2.0.0) ; extra == "map-visualization"
-Requires-Dist: imap-data-access (>=0.10.1)
+Requires-Dist: imap-data-access (>=0.21.0)
 Requires-Dist: mypy (==1.10.1) ; extra == "dev"
 Requires-Dist: netcdf4 (>=1.7.2,<2.0.0) ; extra == "test"
 Requires-Dist: numpy (<=3)
@@ -41,6 +41,7 @@ Requires-Dist: pre-commit (>=3.3.3,<4.0.0) ; extra == "dev"
 Requires-Dist: pydata-sphinx-theme ; extra == "doc"
 Requires-Dist: pytest (>=6.2.5) ; extra == "test"
 Requires-Dist: pytest-cov (>=4.0.0,<5.0.0) ; extra == "test"
+Requires-Dist: pytest-xdist (>=3.2,<4.0) ; extra == "test"
 Requires-Dist: requests (>=2.32.3,<3.0.0) ; extra == "test"
 Requires-Dist: ruff (==0.2.1) ; extra == "dev"
 Requires-Dist: sammi-cdf (>=1.0,<2.0)