dkist-processing-cryonirsp 1.4.14__py3-none-any.whl → 1.4.16rc1__py3-none-any.whl

dkist_processing_cryonirsp/tests/local_trial_workflows/l0_cals_only.py

@@ -127,7 +127,7 @@ def tag_linearized_inputs_task(suffix: str):
                 )
             for file in input_file_list:
                 logger.info(f"Found {file}")
-                self.tag(path=file, tags=[CryonirspTag.linearized(), CryonirspTag.frame()])
+                self.tag(path=file, tags=CryonirspTag.linearized_frame())
             # Update the arm_id constant, as it is derived in linearity processing
             with fits.open(file) as hdul:
                 if len(hdul) == 1:
@@ -189,7 +189,6 @@ def CI_workflow(
     manual_processing_run: ManualProcessing,
     load_beam_boundaries: bool = False,
     load_dark: bool = False,
-    load_lamp: bool = False,
     load_solar: bool = False,
     load_inst_pol: bool = False,
 ) -> None:
@@ -205,12 +204,6 @@ def CI_workflow(
         manual_processing_run.run_task(task=DarkCalibration)
         manual_processing_run.run_task(task=SaveDarkCal)
 
-    if load_lamp:
-        manual_processing_run.run_task(task=LoadLampCal)
-    else:
-        manual_processing_run.run_task(task=LampGainCalibration)
-        manual_processing_run.run_task(task=SaveLampCal)
-
     if load_solar:
         manual_processing_run.run_task(task=LoadSolarCal)
     else:
@@ -365,7 +358,7 @@ def main(
             if not skip_saving_parse:
                 manual_processing_run.run_task(
                     task=save_parsing_task(
-                        tag_list=[CryonirspTag.linearized(), CryonirspTag.frame()],
+                        tag_list=CryonirspTag.linearized_frame(),
                         save_file="linearized_parsing.asdf",
                     )
                 )
@@ -394,7 +387,6 @@ def main(
                 manual_processing_run,
                 load_beam_boundaries=load_beam_boundaries,
                 load_dark=load_dark,
-                load_lamp=load_lamp,
                 load_solar=load_solar,
                 load_inst_pol=load_inst_pol,
             )

dkist_processing_cryonirsp/tests/local_trial_workflows/l0_to_l1.py

@@ -144,7 +144,7 @@ def tag_linearized_inputs_task(suffix: str):
                 )
             for file in input_file_list:
                 logger.info(f"Found {file}")
-                self.tag(path=file, tags=[CryonirspTag.linearized(), CryonirspTag.frame()])
+                self.tag(path=file, tags=CryonirspTag.linearized_frame())
             # Update the arm_id constant, as it is derived in linearity processing
             with fits.open(file) as hdul:
                 if len(hdul) == 1:
@@ -165,7 +165,12 @@ class ShowPolMode(CryonirspTaskBase):
 class ShowExposureConditions(CryonirspTaskBase):
     def run(self) -> None:
         logger.info(f"{self.constants.dark_exposure_conditions_list = }")
-        logger.info(f"{self.constants.lamp_gain_exposure_conditions_list = }")
+        try:
+            logger.info(f"{self.constants.lamp_gain_exposure_conditions_list = }")
+        except KeyError:
+            logger.info(
+                f"Lamp gain exposure conditions list does not exist. This is expected for CRYO-NIRSP CI."
+            )
         logger.info(f"{self.constants.solar_gain_exposure_conditions_list = }")
         if self.constants.correct_for_polarization:
             try:
@@ -207,7 +212,6 @@ def CI_workflow(
     manual_processing_run: ManualProcessing,
     load_beam_boundaries: bool = False,
     load_dark: bool = False,
-    load_lamp: bool = False,
     load_solar: bool = False,
     load_inst_pol: bool = False,
 ) -> None:
@@ -223,12 +227,6 @@ def CI_workflow(
         manual_processing_run.run_task(task=DarkCalibration)
         manual_processing_run.run_task(task=SaveDarkCal)
 
-    if load_lamp:
-        manual_processing_run.run_task(task=LoadLampCal)
-    else:
-        manual_processing_run.run_task(task=LampGainCalibration)
-        manual_processing_run.run_task(task=SaveLampCal)
-
     if load_solar:
         manual_processing_run.run_task(task=LoadSolarCal)
     else:
@@ -390,7 +388,7 @@ def main(
             if not skip_saving_parse:
                 manual_processing_run.run_task(
                     task=save_parsing_task(
-                        tag_list=[CryonirspTag.linearized(), CryonirspTag.frame()],
+                        tag_list=CryonirspTag.linearized_frame(),
                         save_file="linearized_parsing.asdf",
                     )
                 )
@@ -425,7 +423,6 @@ def main(
                 manual_processing_run,
                 load_beam_boundaries=load_beam_boundaries,
                 load_dark=load_dark,
-                load_lamp=load_lamp,
                 load_solar=load_solar,
                 load_inst_pol=load_inst_pol,
             )

dkist_processing_cryonirsp/tests/local_trial_workflows/local_trial_helpers.py

@@ -222,7 +222,7 @@ class SaveGeometricCal(WorkflowTaskBase):
         )
         path_list += list(
             self.read(
-                tags=[CryonirspTag.task_geometric_sepectral_shifts(), CryonirspTag.intermediate()]
+                tags=[CryonirspTag.task_geometric_spectral_shifts(), CryonirspTag.intermediate()]
             )
         )
         path_list += list(

dkist_processing_cryonirsp/tests/test_bad_pixel_maps.py

@@ -81,9 +81,8 @@ def compute_bad_pixel_map_task(
             task.write(
                 data=hdul,
                 tags=[
-                    CryonirspTag.linearized(),
+                    CryonirspTag.linearized_frame(),
                     CryonirspTag.task_solar_gain(),
-                    CryonirspTag.frame(),
                 ],
                 encoder=fits_hdulist_encoder,
             )

dkist_processing_cryonirsp/tests/test_ci_beam_boundaries.py

@@ -5,6 +5,7 @@ import pytest
 from astropy.io import fits
 from dkist_header_validator import spec122_validator
 from dkist_processing_common._util.scratch import WorkflowFileSystem
+from dkist_processing_common.codecs.fits import fits_array_encoder
 from dkist_processing_common.codecs.fits import fits_hdulist_encoder
 from dkist_processing_common.tests.conftest import FakeGQLClient
 
@@ -40,9 +41,10 @@ def compute_beam_boundaries_task(
             param_class = cryonirsp_testing_parameters_factory(param_path=tmp_path)
             assign_input_dataset_doc_to_task(task, param_class())
             # Create fake bad pixel map
-            task.intermediate_frame_write_arrays(
-                arrays=np.zeros(array_shape[1:]),
-                task_tag=CryonirspTag.task_bad_pixel_map(),
+            task.write(
+                data=np.zeros(array_shape[1:]),
+                tags=[CryonirspTag.intermediate_frame(), CryonirspTag.task_bad_pixel_map()],
+                encoder=fits_array_encoder,
             )
             start_time = datetime.now()
             ds = CryonirspHeadersValidCISolarGainFrames(
@@ -73,9 +75,8 @@ def compute_beam_boundaries_task(
             task.write(
                 data=hdul,
                 tags=[
-                    CryonirspTag.linearized(),
+                    CryonirspTag.linearized_frame(),
                     CryonirspTag.task_solar_gain(),
-                    CryonirspTag.frame(),
                 ],
                 encoder=fits_hdulist_encoder,
             )

dkist_processing_cryonirsp/tests/test_ci_science.py

@@ -9,6 +9,7 @@ from astropy.time import Time
 from astropy.time import TimeDelta
 from dkist_header_validator import spec122_validator
 from dkist_processing_common._util.scratch import WorkflowFileSystem
+from dkist_processing_common.codecs.fits import fits_array_encoder
 from dkist_processing_common.codecs.fits import fits_hdulist_encoder
 from dkist_processing_common.tests.conftest import FakeGQLClient
 
@@ -73,14 +74,16 @@ def ci_science_calibration_task(
             assign_input_dataset_doc_to_task(task, param_class())
 
             # Need a beam boundary file
-            task.intermediate_frame_write_arrays(
-                arrays=np.array([0, intermediate_shape[0], 0, intermediate_shape[1]]),
-                task_tag=CryonirspTag.task_beam_boundaries(),
-                beam=1,
+            task.write(
+                data=np.array([0, intermediate_shape[0], 0, intermediate_shape[1]]),
+                tags=[CryonirspTag.intermediate_frame(beam=1), CryonirspTag.task_beam_boundaries()],
+                encoder=fits_array_encoder,
             )
             # Create fake bad pixel map
-            task.intermediate_frame_write_arrays(
-                arrays=np.zeros(array_shape[1:]), task_tag=CryonirspTag.task_bad_pixel_map()
+            task.write(
+                data=np.zeros(array_shape[1:]),
+                tags=[CryonirspTag.intermediate_frame(), CryonirspTag.task_bad_pixel_map()],
+                encoder=fits_array_encoder,
             )
 
             # Create fake demodulation matrices
@@ -91,20 +94,22 @@ def ci_science_calibration_task(
             task.write(
                 data=demod_hdul,
                 tags=[
-                    CryonirspTag.intermediate(),
-                    CryonirspTag.frame(),
+                    CryonirspTag.intermediate_frame(beam=1),
                     CryonirspTag.task_demodulation_matrices(),
-                    CryonirspTag.beam(1),
                 ],
                 encoder=fits_hdulist_encoder,
             )
 
             # Create fake dark intermediate arrays
-            task.intermediate_frame_write_arrays(
-                all_zeros,
-                beam=1,
-                task_tag=CryonirspTag.task_dark(),
-                exposure_conditions=exposure_conditions,
+            task.write(
+                data=all_zeros,
+                tags=[
+                    CryonirspTag.intermediate_frame(
+                        beam=1, exposure_conditions=exposure_conditions
+                    ),
+                    CryonirspTag.task_dark(),
+                ],
+                encoder=fits_array_encoder,
             )
 
             # Create fake lamp and solar gain intermediate arrays
@@ -113,10 +118,8 @@ def ci_science_calibration_task(
                 task.write(
                     data=gain_hdul,
                     tags=[
-                        CryonirspTag.intermediate(),
-                        CryonirspTag.frame(),
+                        CryonirspTag.intermediate_frame(beam=1),
                         CryonirspTag.task_lamp_gain(),
-                        CryonirspTag.beam(1),
                         CryonirspTag.modstate(modstate),
                     ],
                     encoder=fits_hdulist_encoder,
@@ -124,10 +127,8 @@ def ci_science_calibration_task(
                 task.write(
                     data=gain_hdul,
                     tags=[
-                        CryonirspTag.intermediate(),
-                        CryonirspTag.frame(),
+                        CryonirspTag.intermediate_frame(beam=1),
                         CryonirspTag.task_solar_gain(),
-                        CryonirspTag.beam(1),
                         CryonirspTag.modstate(modstate),
                     ],
                     encoder=fits_hdulist_encoder,
@@ -172,9 +173,9 @@ def ci_science_calibration_task(
                                     CryonirspTag.scan_step(scan_step),
                                     CryonirspTag.map_scan(map_scan),
                                     CryonirspTag.modstate(modstate),
-                                    CryonirspTag.linearized(),
-                                    CryonirspTag.frame(),
-                                    CryonirspTag.exposure_conditions(exposure_conditions),
+                                    CryonirspTag.linearized_frame(
+                                        exposure_conditions=exposure_conditions
+                                    ),
                                     CryonirspTag.meas_num(meas_num),
                                 ],
                                 encoder=fits_hdulist_encoder,
@@ -283,7 +284,7 @@ def test_ci_science_calibration_task(ci_science_calibration_task, mocker):
             data = json.load(f)
             assert isinstance(data, dict)
             assert data["total_frames"] == task.scratch.count_all(
-                tags=[CryonirspTag.linearized(), CryonirspTag.frame(), CryonirspTag.task_observe()]
+                tags=[CryonirspTag.linearized_frame(), CryonirspTag.task_observe()]
             )
 
 

dkist_processing_cryonirsp/tests/test_cryo_base.py

@@ -2,14 +2,12 @@ import numpy as np
 import pytest
 from astropy.io import fits
 from dkist_processing_common._util.scratch import WorkflowFileSystem
+from dkist_processing_common.codecs.fits import fits_array_encoder
 from dkist_processing_common.codecs.fits import fits_hdu_decoder
 
+from dkist_processing_cryonirsp.codecs.fits import cryo_fits_array_decoder
 from dkist_processing_cryonirsp.models.tags import CryonirspTag
-from dkist_processing_cryonirsp.models.task_name import CryonirspTaskName
 from dkist_processing_cryonirsp.tasks.cryonirsp_base import CryonirspTaskBase
-from dkist_processing_cryonirsp.tasks.mixin.intermediate_frame import (
-    IntermediateFrameMixin,
-)
 from dkist_processing_cryonirsp.tests.conftest import cryonirsp_testing_parameters_factory
 from dkist_processing_cryonirsp.tests.conftest import CryonirspConstantsDb
 
@@ -24,7 +22,7 @@ WAVE = 1082.0
 def cryo_science_task(
     tmp_path, recipe_run_id, assign_input_dataset_doc_to_task, init_cryonirsp_constants_db
 ):
-    class Task(CryonirspTaskBase, IntermediateFrameMixin):
+    class Task(CryonirspTaskBase):
         def run(self):
             ...
 
@@ -54,21 +52,27 @@ def cryo_science_task(
 def test_write_intermediate_arrays(cryo_science_task):
     """
     Given: A CryonirspTaskBase task
-    When: Using the helper to write a single intermediate array
+    When: Writing a single intermediate array
     Then: The array is written and tagged correctly
     """
     data = np.random.random((10, 10))
     head = fits.Header()
     head["TEST"] = "foo"
-    cryo_science_task.intermediate_frame_write_arrays(
-        arrays=data, headers=head, beam=1, map_scan=2, scan_step=3, task="BAR"
+    cryo_science_task.write(
+        data=data,
+        header=head,
+        tags=[
+            CryonirspTag.intermediate_frame(beam=1),
+            CryonirspTag.map_scan(2),
+            CryonirspTag.scan_step(3),
+            CryonirspTag.task("BAR"),
+        ],
+        encoder=fits_array_encoder,
     )
     loaded_list = list(
         cryo_science_task.read(
             tags=[
-                CryonirspTag.intermediate(),
-                CryonirspTag.frame(),
-                CryonirspTag.beam(1),
+                CryonirspTag.intermediate_frame(beam=1),
                 CryonirspTag.map_scan(2),
                 CryonirspTag.scan_step(3),
                 CryonirspTag.task("BAR"),
@@ -85,7 +89,7 @@ def test_write_intermediate_arrays(cryo_science_task):
 def test_write_intermediate_arrays_task_tag(cryo_science_task):
     """
     Given: A CryonirspTaskBase task
-    When: Using the helper to write a single intermediate array with a formatted task tag input arg
+    When: Writing a single intermediate array with a formatted task tag input arg
     Then: The array is written and tagged correctly
     """
     data = np.random.random((10, 10))
@@ -93,8 +97,11 @@ def test_write_intermediate_arrays_task_tag(cryo_science_task):
     head["TEST"] = "foo"
 
     # bad_pixel_map chosen for no particular reason
-    cryo_science_task.intermediate_frame_write_arrays(
-        arrays=data, headers=head, task_tag=CryonirspTag.task_bad_pixel_map()
+    cryo_science_task.write(
+        data=data,
+        header=head,
+        tags=[CryonirspTag.intermediate_frame(), CryonirspTag.task_bad_pixel_map()],
+        encoder=fits_array_encoder,
     )
     loaded_list = list(
         cryo_science_task.read(
@@ -108,41 +115,27 @@ def test_write_intermediate_arrays_task_tag(cryo_science_task):
     assert hdu.header["TEST"] == "foo"
 
 
-def test_write_intermediate_arrays_task_collisions(cryo_science_task):
-    """
-    Given: A CryonirspTaskBase task
-    When: Using the helper but providing invalid `task` or `task_tag` inputs
-    Then: An error is raised
-    """
-    data = np.random.random((10, 10))
-
-    # Test both given
-    with pytest.raises(ValueError, match="Cannot specify"):
-        cryo_science_task.intermediate_frame_write_arrays(
-            arrays=data, task_tag=CryonirspTag.task_bad_pixel_map(), task="DARK"
-        )
-
-    # Test neither given
-    with pytest.raises(ValueError, match="Must specify"):
-        cryo_science_task.intermediate_frame_write_arrays(arrays=data)
-
-
 def test_write_intermediate_arrays_none_header(cryo_science_task):
     """
     Given: A CryonirspTaskBase task
-    When: Using the helper to write a single intermediate array with no header
+    When: Writing a single intermediate array with no header
     Then: The array is written and tagged correctly
     """
     data = np.random.random((10, 10))
-    cryo_science_task.intermediate_frame_write_arrays(
-        arrays=data, headers=None, beam=1, map_scan=2, scan_step=3, task="BAR"
+    cryo_science_task.write(
+        data=data,
+        tags=[
+            CryonirspTag.intermediate_frame(beam=1),
+            CryonirspTag.map_scan(2),
+            CryonirspTag.scan_step(3),
+            CryonirspTag.task("BAR"),
+        ],
+        encoder=fits_array_encoder,
     )
     loaded_list = list(
         cryo_science_task.read(
             tags=[
-                CryonirspTag.intermediate(),
-                CryonirspTag.frame(),
-                CryonirspTag.beam(1),
+                CryonirspTag.intermediate_frame(beam=1),
                 CryonirspTag.map_scan(2),
                 CryonirspTag.scan_step(3),
                 CryonirspTag.task("BAR"),
@@ -159,7 +152,7 @@ def test_write_intermediate_arrays_none_header(cryo_science_task):
 def cryo_science_task_with_tagged_intermediates(
     recipe_run_id, tmpdir_factory, init_cryonirsp_constants_db
 ):
-    class Task(CryonirspTaskBase, IntermediateFrameMixin):
+    class Task(CryonirspTaskBase):
         def run(self):
             ...
 
@@ -179,7 +172,7 @@ def cryo_science_task_with_tagged_intermediates(
                 hdul = fits.HDUList([fits.PrimaryHDU(data=np.ones((2, 2)) * i)])
                 fname = task.scratch.workflow_base_path / f"file{i}.fits"
                 hdul.writeto(fname)
-                task.tag(fname, tags + [CryonirspTag.intermediate(), CryonirspTag.frame()])
+                task.tag(fname, tags + CryonirspTag.intermediate_frame())
 
             yield task, tag_names, tag_vals
         finally:
@@ -189,7 +182,7 @@ def cryo_science_task_with_tagged_intermediates(
 def test_load_intermediate_arrays(cryo_science_task_with_tagged_intermediates):
     """
     Given: A task with tagged intermediate frames
-    When: Using intermediate frame loaders to grab the intermediate frames
+    When: Reading intermediate frames
     Then: The correct arrays are returned
     """
     task, tag_names, tag_vals = cryo_science_task_with_tagged_intermediates
@@ -197,6 +190,11 @@ def test_load_intermediate_arrays(cryo_science_task_with_tagged_intermediates):
         [getattr(CryonirspTag, n)(v) for n, v in zip(nl, vl)] for nl, vl in zip(tag_names, tag_vals)
     ]
     for i, tags in enumerate(tag_list_list):
-        arrays = list(task.intermediate_frame_load_intermediate_arrays(tags=tags))
+        arrays = list(
+            task.read(
+                tags=[CryonirspTag.intermediate_frame(), tags],
+                decoder=cryo_fits_array_decoder,
+            )
+        )
         assert len(arrays) == 1
         np.testing.assert_equal(arrays[0], np.ones((2, 2)) * i)

dkist_processing_cryonirsp/tests/test_dark.py

@@ -5,6 +5,7 @@ import pytest
 from astropy.io import fits
 from dkist_header_validator import spec122_validator
 from dkist_processing_common._util.scratch import WorkflowFileSystem
+from dkist_processing_common.codecs.fits import fits_array_encoder
 from dkist_processing_common.codecs.fits import fits_hdulist_encoder
 from dkist_processing_common.tests.conftest import FakeGQLClient
 
@@ -65,10 +66,13 @@ def sp_dark_calibration_task(
                         spectral_starting_pixel + illuminated_beam_shape[1],
                     ]
                 )
-                task.intermediate_frame_write_arrays(
-                    arrays=beam_boundaries,
-                    task_tag=CryonirspTag.task_beam_boundaries(),
-                    beam=beam,
+                task.write(
+                    data=beam_boundaries,
+                    tags=[
+                        CryonirspTag.intermediate_frame(beam=beam),
+                        CryonirspTag.task_beam_boundaries(),
+                    ],
+                    encoder=fits_array_encoder,
                 )
 
             ds = CryonirspHeadersValidDarkFrames(
@@ -91,10 +95,8 @@ def sp_dark_calibration_task(
                     task.write(
                         data=hdul,
                         tags=[
-                            CryonirspTag.linearized(),
-                            CryonirspTag.frame(),
+                            CryonirspTag.linearized_frame(exposure_conditions=condition),
                             CryonirspTag.task_dark(),
-                            CryonirspTag.exposure_conditions(condition),
                         ],
                         encoder=fits_hdulist_encoder,
                     )
@@ -139,10 +141,10 @@ def ci_dark_calibration_task(
             assign_input_dataset_doc_to_task(task, param_class())
 
             # Need a beam boundary file
-            task.intermediate_frame_write_arrays(
-                arrays=np.array([0, 10, 0, 10]),
-                task_tag=CryonirspTag.task_beam_boundaries(),
-                beam=1,
+            task.write(
+                data=np.array([0, 10, 0, 10]),
+                tags=[CryonirspTag.intermediate_frame(beam=1), CryonirspTag.task_beam_boundaries()],
+                encoder=fits_array_encoder,
             )
 
             ds = CryonirspHeadersValidDarkFrames(
@@ -164,10 +166,8 @@ def ci_dark_calibration_task(
                     task.write(
                         data=hdul,
                         tags=[
-                            CryonirspTag.linearized(),
-                            CryonirspTag.frame(),
+                            CryonirspTag.linearized_frame(exposure_conditions=condition),
                             CryonirspTag.task_dark(),
-                            CryonirspTag.exposure_conditions(condition),
                         ],
                         encoder=fits_hdulist_encoder,
                     )
@@ -201,9 +201,7 @@ def test_sp_dark_calibration_task(sp_dark_calibration_task, mocker):
                 task.read(
                     tags=[
                         CryonirspTag.task_dark(),
-                        CryonirspTag.intermediate(),
-                        CryonirspTag.frame(),
-                        CryonirspTag.beam(b + 1),
+                        CryonirspTag.intermediate_frame(beam=b + 1),
                         CryonirspTag.exposure_conditions(condition),
                     ]
                 )
@@ -217,8 +215,7 @@ def test_sp_dark_calibration_task(sp_dark_calibration_task, mocker):
     unused_time_read = task.read(
         tags=[
             CryonirspTag.task_dark(),
-            CryonirspTag.intermediate(),
-            CryonirspTag.frame(),
+            CryonirspTag.intermediate_frame(),
             CryonirspTag.exposure_conditions(unused_condition),
         ]
     )
@@ -230,7 +227,7 @@ def test_sp_dark_calibration_task(sp_dark_calibration_task, mocker):
             data = json.load(f)
             assert isinstance(data, dict)
             assert data["total_frames"] == task.scratch.count_all(
-                tags=[CryonirspTag.linearized(), CryonirspTag.frame(), CryonirspTag.task_dark()]
+                tags=[CryonirspTag.linearized_frame(), CryonirspTag.task_dark()]
             )
             assert data["frames_not_used"] == 3
 
@@ -253,10 +250,7 @@ def test_ci_dark_calibration_task(ci_dark_calibration_task, mocker):
             task.read(
                 tags=[
                     CryonirspTag.task_dark(),
-                    CryonirspTag.beam(1),
-                    CryonirspTag.intermediate(),
-                    CryonirspTag.frame(),
-                    CryonirspTag.exposure_conditions(condition),
+                    CryonirspTag.intermediate_frame(beam=1, exposure_conditions=condition),
                 ]
             )
         )
@@ -269,9 +263,7 @@ def test_ci_dark_calibration_task(ci_dark_calibration_task, mocker):
     unused_time_read = task.read(
         tags=[
             CryonirspTag.task_dark(),
-            CryonirspTag.intermediate(),
-            CryonirspTag.frame(),
-            CryonirspTag.exposure_conditions(unused_condition),
+            CryonirspTag.intermediate_frame(exposure_conditions=unused_condition),
         ]
     )
     assert len(list(unused_time_read)) == 0
@@ -282,6 +274,6 @@ def test_ci_dark_calibration_task(ci_dark_calibration_task, mocker):
             data = json.load(f)
             assert isinstance(data, dict)
             assert data["total_frames"] == task.scratch.count_all(
-                tags=[CryonirspTag.linearized(), CryonirspTag.frame(), CryonirspTag.task_dark()]
+                tags=[CryonirspTag.linearized_frame(), CryonirspTag.task_dark()]
             )
             assert data["frames_not_used"] == 3