Merge pull request #46 from lsst/tickets/DM-47952

DM-47952: Add Real/Bogus to drp_pipe

bin.src/ci_imsim_run.py

@@ -56,6 +56,15 @@ class ImsimDefineVisits(DefineVisits):
     collectionsName = f"{INSTRUMENT_NAME}/raw/all"
 
 
+@ciRunner.register("import_external_pretrained_models", index_command := index_command + 1)
+class ImsimButlerImportPretrainedModels(ButlerImport):
+    dataLocation = TESTDATA_DIR
+
+    @property
+    def importFileLocation(self) -> str:
+        return os.path.join(self.runner.pkgRoot, "resources", "external_pretrained_models.yaml")
+
+
 @ciRunner.register("import_external", index_command := index_command + 1)
 class ImsimBaseButlerImport(ButlerImport):
     dataLocation = TESTDATA_DIR

resources/external.yaml

@@ -63,6 +63,7 @@ data:
   - LSSTCam-imSim/refcats
   - LSSTCam-imSim/truth_summary
   - LSSTCam-imSim/red_galaxies_cosmodc2
+  - pretrained_models/tac_cnn_dc2_2024-11-14
 - type: dataset_type
   name: bias
   dimensions:

resources/external_pretrained_models.yaml

@@ -0,0 +1,26 @@
+description: Butler Data Repository Export
+version: 1.0.2
+universe_version: 7
+universe_namespace: daf_butler
+data:
+- type: collection
+  collection_type: RUN
+  name: pretrained_models/tac_cnn_dc2_2024-11-14
+  host: null
+  timespan_begin: null
+  timespan_end: null
+- type: dataset_type
+  name: pretrainedModelPackage
+  dimensions: []
+  storage_class: NNModelPackagePayload
+  is_calibration: false
+- type: dataset
+  dataset_type: pretrainedModelPackage
+  run: pretrained_models/tac_cnn_dc2_2024-11-14
+  records:
+  - dataset_id:
+    - !uuid '3c2f1a51-0f8c-424d-bd6f-e0138cedce22'
+    data_id:
+    - {}
+    path: pretrained_models/tac_cnn_dc2_2024-11-14/pretrainedModelPackage/pretrainedModelPackage_pretrained_models_tac_cnn_dc2_2024-11-14.zip
+    formatter: lsst.meas.transiNet.modelPackages.formatters.NNModelPackageFormatter