Merge pull request #918 from lsst/tickets/DM-38454

DM-38454: Add the ModelPackage storage class.
lsst · Dec 15, 2023 · 0e097e9 · 0e097e9
2 parents cc65914 + 964113a
commit 0e097e9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/doc/changes/DM-38454.feature.rst b/doc/changes/DM-38454.feature.rst
@@ -0,0 +1 @@
+Added new storageClass and formatter for NNModelPackagePayload -- an interface between butler and pretrained neural networks, currently implemented in pytorch.
diff --git a/python/lsst/daf/butler/configs/datastores/formatters.yaml b/python/lsst/daf/butler/configs/datastores/formatters.yaml
@@ -88,3 +88,4 @@ SpectractorFitParameters: lsst.atmospec.formatters.SpectractorFitParametersForma
 ScarletModelData: lsst.daf.butler.formatters.json.JsonFormatter
 MetricMeasurementBundle: lsst.daf.butler.formatters.json.JsonFormatter
 MultipleCellCoadd: lsst.cell_coadds.CellCoaddFitsFormatter
+NNModelPackagePayload: lsst.meas.transiNet.modelPackages.NNModelPackageFormatter
diff --git a/python/lsst/daf/butler/configs/storageClasses.yaml b/python/lsst/daf/butler/configs/storageClasses.yaml
@@ -411,3 +411,5 @@ storageClasses:
     pytype: lsst.analysis.tools.interfaces.MetricMeasurementBundle
   MultipleCellCoadd:
     pytype: lsst.cell_coadds.MultipleCellCoadd
+  NNModelPackagePayload:
+    pytype: lsst.meas.transiNet.modelPackages.formatters.NNModelPackagePayload