Merge pull request #1056 from lsst/tickets/DM-45541

DM-45541: Add QPEnsemble and PZModel to storage classes and formatters
lsst · Aug 15, 2024 · 41ffdbb · 41ffdbb
2 parents 6280842 + cddaa8b
commit 41ffdbb
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/doc/changes/DM-45541.feature.rst b/doc/changes/DM-45541.feature.rst
@@ -0,0 +1 @@
+Added ``QPEnsemble`` and ``PZModel`` to ``datastores/formatters.yaml`` and ``storageClasses.yaml`` to enable storage of the machine learning models used by photo-z algorithms as well as the photo-z estimates produced by those algorithms.
diff --git a/python/lsst/daf/butler/configs/datastores/formatters.yaml b/python/lsst/daf/butler/configs/datastores/formatters.yaml
@@ -92,3 +92,5 @@ MultipleCellCoadd: lsst.cell_coadds.CellCoaddFitsFormatter
 NNModelPackagePayload: lsst.meas.transiNet.modelPackages.NNModelPackageFormatter
 Timespan: lsst.daf.butler.formatters.json.JsonFormatter
 RegionTimeInfo: lsst.daf.butler.formatters.json.JsonFormatter
+QPEnsemble: lsst.meas.pz.qp_formatter.QPFormatter
+PZModel: lsst.meas.pz.model_formatter.ModelFormatter
diff --git a/python/lsst/daf/butler/configs/storageClasses.yaml b/python/lsst/daf/butler/configs/storageClasses.yaml
@@ -419,3 +419,7 @@ storageClasses:
     pytype: lsst.daf.butler.Timespan
   RegionTimeInfo:
     pytype: lsst.pipe.base.utils.RegionTimeInfo
+  QPEnsemble:
+    pytype: qp.Ensemble
+  PZModel:
+    pytype: rail.core.model.Model