dkist-processing-common 11.3.1rc1__tar.gz → 11.4.0rc1__tar.gz

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dkist-processing-common
-Version: 11.3.1rc1
+Version: 11.4.0rc1
 Summary: Common task classes used by the DKIST science data processing pipelines
 Author-email: NSO / AURA <dkistdc@nso.edu>
 License: BSD-3-Clause
@@ -10,14 +10,14 @@ Project-URL: Documentation, https://docs.dkist.nso.edu/projects/common
 Project-URL: Help, https://nso.atlassian.net/servicedesk/customer/portal/5
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Requires-Python: >=3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.12
 Description-Content-Type: text/x-rst
 Requires-Dist: asdf<4.0.0,>=3.5.0
 Requires-Dist: astropy>=7.0.0
 Requires-Dist: dkist-fits-specifications<5.0,>=4.0.0
 Requires-Dist: dkist-header-validator<6.0,>=5.0.0
-Requires-Dist: dkist-processing-core==5.1.1
+Requires-Dist: dkist-processing-core==5.2.0rc1
 Requires-Dist: dkist-processing-pac<4.0,>=3.1
 Requires-Dist: dkist-service-configuration<3.0,>=2.0.2
 Requires-Dist: dkist-spectral-lines<4.0,>=3.0.0
@@ -28,7 +28,7 @@ Requires-Dist: sqids==0.5.1
 Requires-Dist: matplotlib>=3.4
 Requires-Dist: moviepy>=2.0.0
 Requires-Dist: numpy>=1.26.4
-Requires-Dist: object-clerk==0.1.2
+Requires-Dist: object-clerk==1.0.0
 Requires-Dist: pandas>=1.4.2
 Requires-Dist: pillow>=10.2.0
 Requires-Dist: pydantic>=2.0
@@ -36,7 +36,7 @@ Requires-Dist: redis==4.6.0
 Requires-Dist: requests>=2.23
 Requires-Dist: scipy>=1.15.1
 Requires-Dist: sunpy>=3.0.0
-Requires-Dist: talus==1.1.0
+Requires-Dist: talus==1.3.4
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-xdist; extra == "test"

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/bitbucket-pipelines.yml

@@ -1,4 +1,4 @@
-image: python:3.11
+image: python:3.12
 definitions:
   services:
     redis:

dkist_processing_common-11.4.0rc1/changelog/266.misc.rst

@@ -0,0 +1 @@
+Update dkist-processing-core to 5.2.0 which includes upgrades to airflow 2.11.0 and requires Python 3.12+.

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/models/constants.py

@@ -5,7 +5,7 @@ Contains names of database entries and Base class for an object that simplifies
 accessing the database (tab completion, etc.)
 """
 
-from enum import StrEnum
+from enum import Enum
 from string import ascii_uppercase
 
 from sqids import Sqids
@@ -13,13 +13,13 @@ from sqids import Sqids
 from dkist_processing_common._util.constants import ConstantsDb
 
 
-class BudName(StrEnum):
+class BudName(str, Enum):
     """Controlled list of names for constant stems (buds)."""
 
+    recipe_run_id = "RECIPE_RUN_ID"
     instrument = "INSTRUMENT"
     num_cs_steps = "NUM_CS_STEPS"
     num_modstates = "NUM_MODSTATES"
-    retarder_name = "RETARDER_NAME"
     proposal_id = "PROPOSAL_ID"
     contributing_proposal_ids = "CONTRIBUTING_PROPOSAL_IDS"
     experiment_id = "EXPERIMENT_ID"
@@ -33,6 +33,7 @@ class BudName(StrEnum):
     dark_exposure_times = "DARK_EXPOSURE_TIMES"
     dark_readout_exp_times = "DARK_READOUT_EXP_TIMES"
     wavelength = "WAVELENGTH"
+    retarder_name = "RETARDER_NAME"
 
 
 class ConstantsBase:
@@ -62,7 +63,7 @@ class ConstantsBase:
     Parameters
     ----------
     recipe_run_id
-        The recipe_run_id
+        Thew recipe_run_id
     task_name
         The task_name
     """
@@ -90,31 +91,6 @@ class ConstantsBase:
         """Define the dataset_id constant."""
         return Sqids(min_length=6, alphabet=ascii_uppercase).encode([self._recipe_run_id])
 
-    @property
-    def stokes_params(self) -> [str]:
-        """Return the list of stokes parameter names."""
-        return ["I", "Q", "U", "V"]
-
-    @property
-    def instrument(self) -> str:
-        """Get the instrument name."""
-        return self._db_dict[BudName.instrument]
-
-    @property
-    def num_cs_steps(self):
-        """Get the number of calibration sequence steps."""
-        return self._db_dict[BudName.num_cs_steps]
-
-    @property
-    def num_modstates(self):
-        """Get the number of modulation states."""
-        return self._db_dict[BudName.num_modstates]
-
-    @property
-    def retarder_name(self):
-        """Get the retarder name."""
-        return self._db_dict[BudName.retarder_name]
-
     @property
     def proposal_id(self) -> str:
         """Get the proposal_id constant."""
@@ -129,22 +105,14 @@ class ConstantsBase:
         return proposal_ids
 
     @property
-    def experiment_id(self) -> str:
-        """Get the experiment_id constant."""
-        return self._db_dict[BudName.experiment_id]
-
-    @property
-    def contributing_experiment_ids(self) -> [str]:
-        """Return the list of contributing experiment IDs."""
-        experiment_ids = self._db_dict[BudName.contributing_experiment_ids]
-        if isinstance(experiment_ids, str):
-            return [experiment_ids]
-        return experiment_ids
+    def instrument(self) -> str:
+        """Get the instrument name."""
+        return self._db_dict[BudName.instrument]
 
     @property
     def obs_ip_start_time(self) -> str:
         """Return the start time of the observe IP."""
-        return self._db_dict[BudName.obs_ip_start_time]
+        return self._db_dict[BudName.obs_ip_start_time.value]
 
     @property
     def average_cadence(self) -> float:
@@ -181,7 +149,25 @@ class ConstantsBase:
         """Get a list of readout exp times for all DARK frames."""
         return self._db_dict[BudName.dark_readout_exp_times]
 
+    @property
+    def stokes_params(self) -> [str]:
+        """Return the list of stokes parameter names."""
+        return ["I", "Q", "U", "V"]
+
+    @property
+    def experiment_id(self) -> str:
+        """Get the experiment_id constant."""
+        return self._db_dict[BudName.experiment_id]
+
+    @property
+    def contributing_experiment_ids(self) -> [str]:
+        """Return the list of contributing experiment IDs."""
+        experiment_ids = self._db_dict[BudName.contributing_experiment_ids]
+        if isinstance(experiment_ids, str):
+            return [experiment_ids]
+        return experiment_ids
+
     @property
     def wavelength(self) -> float:
         """Wavelength."""
-        return self._db_dict[BudName.wavelength]
+        return self._db_dict[BudName.wavelength.value]

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/models/fits_access.py

@@ -2,44 +2,11 @@
 
 from __future__ import annotations
 
-from enum import StrEnum
 from pathlib import Path
-from typing import Any
 
 import numpy as np
 from astropy.io import fits
 
-NOT_FOUND_MESSAGE = "_HEADER_KEYWORD_NOT_FOUND"
-
-
-class MetadataKey(StrEnum):
-    """Controlled list of names for FITS metadata header keys."""
-
-    ip_task_type = "IPTASK"  # in L0FitsAccess
-    ip_start_time = "DKIST011"  # in L0FitsAccess
-    ip_end_time = "DKIST012"  # in L0FitsAccess
-    elevation = "ELEV_ANG"
-    azimuth = "TAZIMUTH"
-    table_angle = "TTBLANGL"
-    gos_level3_status = "LVL3STAT"
-    gos_level3_lamp_status = "LAMPSTAT"
-    gos_polarizer_status = "LVL2STAT"
-    gos_polarizer_angle = "POLANGLE"
-    gos_retarder_status = "LVL1STAT"
-    gos_retarder_angle = "RETANGLE"
-    gos_level0_status = "LVL0STAT"
-    time_obs = "DATE-BEG"
-    ip_id = "IP_ID"
-    instrument = "INSTRUME"
-    wavelength = "LINEWAV"
-    proposal_id = "PROP_ID"
-    experiment_id = "EXPER_ID"
-    num_dsps_repeats = "DSPSREPS"
-    current_dsps_repeat = "DSPSNUM"
-    fpa_exposure_time_ms = "XPOSURE"
-    sensor_readout_exposure_time_ms = "TEXPOSUR"
-    num_raw_frames_per_fpa = "NSUMEXP"
-
 
 class FitsAccessBase:
     """
@@ -68,29 +35,6 @@ class FitsAccessBase:
     def __repr__(self):
         return f"{self.__class__.__name__}(hdu={self._hdu!r}, name={self.name!r}, auto_squeeze={self.auto_squeeze})"
 
-    def _set_metadata_key_value(
-        self, key: StrEnum, optional: bool = False, default: Any = NOT_FOUND_MESSAGE
-    ) -> None:
-        """
-        Get the header value and assign it as a metadata key name attribute.
-
-        Parameters
-        ----------
-        key
-            The StrEnum member in attribute_name = fits_keyword structure
-        optional
-            If the keyword is optional
-        default
-            Value for the attribute if the key is not found
-        """
-        if optional:
-            if default != NOT_FOUND_MESSAGE:
-                setattr(self, key.name, self.header.get(key, default))
-            else:
-                setattr(self, key.name, self.header.get(key, key + NOT_FOUND_MESSAGE))
-        else:
-            setattr(self, key.name, self.header[key])
-
     @property
     def data(self) -> np.ndarray:
         """

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/models/tags.py

@@ -1,7 +1,6 @@
 """Components of the Tag model.  Stem + Optional Suffix = Tag."""
 
 from enum import Enum
-from enum import StrEnum
 
 from dkist_processing_common.models.task_name import TaskName
 
@@ -9,7 +8,7 @@ from dkist_processing_common.models.task_name import TaskName
 EXP_TIME_ROUND_DIGITS: int = 6
 
 
-class StemName(StrEnum):
+class StemName(str, Enum):
     """Controlled list of Tag Stems."""
 
     output = "OUTPUT"

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/models/task_name.py

@@ -1,9 +1,9 @@
 """Controlled list of common IP task tag names."""
 
-from enum import StrEnum
+from enum import Enum
 
 
-class TaskName(StrEnum):
+class TaskName(str, Enum):
     """Controlled list of task tag names."""
 
     observe = "OBSERVE"

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/cs_step.py

@@ -101,7 +101,7 @@ class CSStepFlower(Stem):
     """
 
     def __init__(self, max_cs_step_time_sec: float):
-        super().__init__(stem_name=StemName.cs_step)
+        super().__init__(stem_name=StemName.cs_step.value)
         self.max_cs_step_time_sec = max_cs_step_time_sec
 
     def setter(self, fits_obj: L0FitsAccess) -> CSStep | Type[SpilledDirt]:
@@ -148,7 +148,7 @@ class NumCSStepBud(Stem):
     """
 
     def __init__(self, max_cs_step_time_sec: float):
-        super().__init__(stem_name=BudName.num_cs_steps)
+        super().__init__(stem_name=BudName.num_cs_steps.value)
         self.max_cs_step_time_sec = max_cs_step_time_sec
 
     def setter(self, fits_obj: L0FitsAccess) -> CSStep | Type[SpilledDirt]:

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/dsps_repeat.py

@@ -1,7 +1,6 @@
 """Classes supporting the Data Set Parameters Set (DSPS) Repeat parameter."""
 
 from dkist_processing_common.models.constants import BudName
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.models.flower_pot import SpilledDirt
 from dkist_processing_common.models.tags import StemName
 from dkist_processing_common.models.task_name import TaskName
@@ -17,9 +16,9 @@ class TotalDspsRepeatsBud(TaskUniqueBud):
 
     def __init__(self):
         super().__init__(
-            constant_name=BudName.num_dsps_repeats,
-            metadata_key=MetadataKey.num_dsps_repeats,
-            ip_task_types=TaskName.observe,
+            constant_name=BudName.num_dsps_repeats.value,
+            metadata_key="num_dsps_repeats",
+            ip_task_types=TaskName.observe.value,
         )
 
 
@@ -28,7 +27,7 @@ class DspsRepeatNumberFlower(SingleValueSingleKeyFlower):
 
     def __init__(self):
         super().__init__(
-            tag_stem_name=StemName.dsps_repeat, metadata_key=MetadataKey.current_dsps_repeat
+            tag_stem_name=StemName.dsps_repeat.value, metadata_key="current_dsps_repeat"
         )
 
     def setter(self, fits_obj: L0FitsAccess):

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/experiment_id_bud.py

@@ -1,7 +1,6 @@
 """Experiment Id parser."""
 
 from dkist_processing_common.models.constants import BudName
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.parsers.id_bud import ContributingIdsBud
 from dkist_processing_common.parsers.id_bud import IdBud
 
@@ -10,9 +9,7 @@ class ExperimentIdBud(IdBud):
     """Class to create a Bud for the experiment_id."""
 
     def __init__(self):
-        super().__init__(
-            constant_name=BudName.experiment_id, metadata_key=MetadataKey.experiment_id
-        )
+        super().__init__(constant_name=BudName.experiment_id.value, metadata_key="experiment_id")
 
 
 class ContributingExperimentIdsBud(ContributingIdsBud):
@@ -20,5 +17,5 @@ class ContributingExperimentIdsBud(ContributingIdsBud):
 
     def __init__(self):
         super().__init__(
-            stem_name=BudName.contributing_experiment_ids, metadata_key=MetadataKey.experiment_id
+            stem_name=BudName.contributing_experiment_ids.value, metadata_key="experiment_id"
         )

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/id_bud.py

@@ -1,6 +1,5 @@
 """Base classes for ID bud parsing."""
 
-from enum import StrEnum
 from typing import Type
 
 from dkist_processing_common.models.flower_pot import SpilledDirt
@@ -13,21 +12,19 @@ from dkist_processing_common.parsers.unique_bud import TaskUniqueBud
 class IdBud(TaskUniqueBud):
     """Base class for ID buds."""
 
-    def __init__(self, constant_name: str, metadata_key: str | StrEnum):
+    def __init__(self, constant_name, metadata_key):
         super().__init__(
             constant_name=constant_name,
             metadata_key=metadata_key,
-            ip_task_types=TaskName.observe,
+            ip_task_types=TaskName.observe.value,
         )
 
 
 class ContributingIdsBud(Stem):
     """Base class for contributing ID buds."""
 
-    def __init__(self, stem_name: str, metadata_key: str | StrEnum):
+    def __init__(self, stem_name, metadata_key):
         super().__init__(stem_name=stem_name)
-        if isinstance(metadata_key, StrEnum):
-            metadata_key = metadata_key.name
         self.metadata_key = metadata_key
 
     def setter(self, fits_obj: L0FitsAccess) -> str | Type[SpilledDirt]:

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/l0_fits_access.py

@@ -2,7 +2,6 @@
 
 from astropy.io import fits
 
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.parsers.l1_fits_access import L1FitsAccess
 
 
@@ -27,6 +26,6 @@ class L0FitsAccess(L1FitsAccess):
         auto_squeeze: bool = True,
     ):
         super().__init__(hdu=hdu, name=name, auto_squeeze=auto_squeeze)
-        self._set_metadata_key_value(MetadataKey.ip_task_type)
-        self._set_metadata_key_value(MetadataKey.ip_start_time)
-        self._set_metadata_key_value(MetadataKey.ip_end_time)
+        self.ip_task_type: str = self.header["IPTASK"]
+        self.ip_start_time: str = self.header["DKIST011"]
+        self.ip_end_time: str = self.header["DKIST012"]

dkist_processing_common-11.4.0rc1/dkist_processing_common/parsers/l1_fits_access.py

@@ -0,0 +1,64 @@
+"""By-frame 214 L1 only header keywords that are not instrument specific."""
+
+from astropy.io import fits
+
+from dkist_processing_common.models.fits_access import FitsAccessBase
+
+
+class L1FitsAccess(FitsAccessBase):
+    """
+    Class defining a fits access object for processed L1 data.
+
+    Parameters
+    ----------
+    hdu
+        The input fits hdu
+    name
+        An optional name to be associated with the hdu
+    auto_squeeze
+        A boolean indicating whether to 'squeeze' out dimensions of size 1
+    """
+
+    def __init__(
+        self,
+        hdu: fits.ImageHDU | fits.PrimaryHDU | fits.CompImageHDU,
+        name: str | None = None,
+        auto_squeeze: bool = False,  # Because L1 data should always have the right form, right?
+    ):
+        super().__init__(hdu=hdu, name=name, auto_squeeze=auto_squeeze)
+
+        self.elevation: float = self.header["ELEV_ANG"]
+        self.azimuth: float = self.header["TAZIMUTH"]
+        self.table_angle: float = self.header["TTBLANGL"]
+        self.gos_level3_status: str = self.header["LVL3STAT"]
+        self.gos_level3_lamp_status: str = self.header["LAMPSTAT"]
+        self.gos_polarizer_status: str = self.header["LVL2STAT"]
+        self.gos_retarder_status: str = self.header["LVL1STAT"]
+        self.gos_level0_status: str = self.header["LVL0STAT"]
+        self.time_obs: str = self.header["DATE-BEG"]
+        self.ip_id: str = self.header["IP_ID"]
+        self.instrument: str = self.header["INSTRUME"]
+        self.wavelength: float = self.header["LINEWAV"]
+        self.proposal_id: str = self.header["PROP_ID"]
+        self.experiment_id: str = self.header["EXPER_ID"]
+        self.num_dsps_repeats: int = self.header["DSPSREPS"]
+        self.current_dsps_repeat: int = self.header["DSPSNUM"]
+        self.fpa_exposure_time_ms: float = self.header["XPOSURE"]
+        self.sensor_readout_exposure_time_ms: float = self.header["TEXPOSUR"]
+        self.num_raw_frames_per_fpa: int = self.header["NSUMEXP"]
+
+    @property
+    def gos_polarizer_angle(self) -> float:
+        """Convert the polarizer angle to a float if possible before returning."""
+        try:
+            return float(self.header["POLANGLE"])
+        except ValueError:
+            return -999  # The angle is only used if the polarizer is in the beam
+
+    @property
+    def gos_retarder_angle(self) -> float:
+        """Convert the retarder angle to a float if possible before returning."""
+        try:
+            return float(self.header["RETANGLE"])
+        except ValueError:
+            return -999  # The angle is only used if the retarder is in the beam

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/near_bud.py

@@ -1,6 +1,5 @@
 """Pre-made flower that reads a single header key from all files and raises a ValueError if the values are not in a supplied range."""
 
-from enum import StrEnum
 from statistics import mean
 from typing import Callable
 
@@ -31,12 +30,10 @@ class NearFloatBud(Stem):
     def __init__(
         self,
         constant_name: str,
-        metadata_key: str | StrEnum,
+        metadata_key: str,
         tolerance: float,
     ):
         super().__init__(stem_name=constant_name)
-        if isinstance(metadata_key, StrEnum):
-            metadata_key = metadata_key.name
         self.metadata_key = metadata_key
         self.tolerance = tolerance
 
@@ -101,7 +98,7 @@ class TaskNearFloatBud(NearFloatBud):
     def __init__(
         self,
         constant_name: str,
-        metadata_key: str | StrEnum,
+        metadata_key: str,
         ip_task_types: str | list[str],
         tolerance: float,
         task_type_parsing_function: Callable = passthrough_header_ip_task,

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/proposal_id_bud.py

@@ -1,7 +1,6 @@
 """Proposal Id parser."""
 
 from dkist_processing_common.models.constants import BudName
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.parsers.id_bud import ContributingIdsBud
 from dkist_processing_common.parsers.id_bud import IdBud
 
@@ -10,7 +9,7 @@ class ProposalIdBud(IdBud):
     """Class to create a Bud for the proposal_id."""
 
     def __init__(self):
-        super().__init__(constant_name=BudName.proposal_id, metadata_key=MetadataKey.proposal_id)
+        super().__init__(constant_name=BudName.proposal_id.value, metadata_key="proposal_id")
 
 
 class ContributingProposalIdsBud(ContributingIdsBud):
@@ -18,5 +17,5 @@ class ContributingProposalIdsBud(ContributingIdsBud):
 
     def __init__(self):
         super().__init__(
-            stem_name=BudName.contributing_proposal_ids, metadata_key=MetadataKey.proposal_id
+            stem_name=BudName.contributing_proposal_ids.value, metadata_key="proposal_id"
         )

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/retarder.py

@@ -1,7 +1,6 @@
 """Bud that parses the name of the retarder used during POLCAL task observations."""
 
 from dkist_processing_common.models.constants import BudName
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.models.task_name import TaskName
 from dkist_processing_common.parsers.unique_bud import TaskUniqueBud
 
@@ -20,9 +19,9 @@ class RetarderNameBud(TaskUniqueBud):
 
     def __init__(self):
         super().__init__(
-            constant_name=BudName.retarder_name,
-            metadata_key=MetadataKey.gos_retarder_status,
-            ip_task_types=TaskName.polcal,
+            constant_name=BudName.retarder_name.value,
+            metadata_key="gos_retarder_status",
+            ip_task_types=TaskName.polcal.value,
         )
 
     def getter(self, key) -> str:

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/single_value_single_key_flower.py

@@ -1,8 +1,5 @@
 """Pre-made flower that produces tag based on a single header key."""
 
-from enum import StrEnum
-
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.models.flower_pot import Stem
 from dkist_processing_common.parsers.l0_fits_access import L0FitsAccess
 
@@ -19,10 +16,8 @@ class SingleValueSingleKeyFlower(Stem):
         The metadata key
     """
 
-    def __init__(self, tag_stem_name: str, metadata_key: str | StrEnum):
+    def __init__(self, tag_stem_name: str, metadata_key: str):
         super().__init__(stem_name=tag_stem_name)
-        if isinstance(metadata_key, StrEnum):
-            metadata_key = metadata_key.name
         self.metadata_key = metadata_key
 
     def setter(self, fits_obj: L0FitsAccess):

{dkist_processing_common-11.3.1rc1 → dkist_processing_common-11.4.0rc1}/dkist_processing_common/parsers/task.py

@@ -4,7 +4,6 @@ from typing import Callable
 from typing import Type
 
 from dkist_processing_common.models.fits_access import FitsAccessBase
-from dkist_processing_common.models.fits_access import MetadataKey
 from dkist_processing_common.models.flower_pot import SpilledDirt
 from dkist_processing_common.models.tags import StemName
 from dkist_processing_common.models.task_name import TaskName
@@ -37,9 +36,9 @@ def parse_header_ip_task_with_gains(fits_obj: FitsAccessBase) -> str:
         and fits_obj.gos_level3_status == "lamp"
         and fits_obj.gos_level3_lamp_status == "on"
     ):
-        return TaskName.lamp_gain
+        return TaskName.lamp_gain.value
     if fits_obj.ip_task_type == "gain" and fits_obj.gos_level3_status == "clear":
-        return TaskName.solar_gain
+        return TaskName.solar_gain.value
 
     # Everything else is unchanged
     return passthrough_header_ip_task(fits_obj)
@@ -60,14 +59,14 @@ def parse_polcal_task_type(fits_obj: Type[FitsAccessBase]) -> str | Type[Spilled
         and fits_obj.gos_retarder_status == "clear"
         and fits_obj.gos_polarizer_status == "clear"
     ):
-        return TaskName.polcal_dark
+        return TaskName.polcal_dark.value
 
     elif (
         fits_obj.gos_level0_status.startswith("FieldStop")
         and fits_obj.gos_retarder_status == "clear"
         and fits_obj.gos_polarizer_status == "clear"
     ):
-        return TaskName.polcal_gain
+        return TaskName.polcal_gain.value
 
     # We don't care about a POLCAL frame that is neither dark nor clear
     return SpilledDirt
@@ -79,7 +78,7 @@ class TaskTypeFlower(SingleValueSingleKeyFlower):
     def __init__(
         self, header_task_parsing_func: Callable[[FitsAccessBase], str] = passthrough_header_ip_task
     ):
-        super().__init__(tag_stem_name=StemName.task, metadata_key=MetadataKey.ip_task_type)
+        super().__init__(tag_stem_name=StemName.task.value, metadata_key="ip_task_type")
         self.header_parsing_function = header_task_parsing_func
 
     def setter(self, fits_obj: FitsAccessBase):
@@ -104,7 +103,7 @@ class PolcalTaskFlower(SingleValueSingleKeyFlower):
     """
 
     def __init__(self):
-        super().__init__(tag_stem_name=StemName.task, metadata_key=MetadataKey.ip_task_type)
+        super().__init__(tag_stem_name=StemName.task.value, metadata_key="ip_task_type")
 
     def setter(self, fits_obj: FitsAccessBase):
         """