Forced photometry update pipeline - missing match stage (#264)

added missing match stage
skyportal · Dec 13, 2023 · dc1a9bd · dc1a9bd
1 parent 7f29ad0
commit dc1a9bd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/kowalski/alert_brokers/alert_broker_ztf.py b/kowalski/alert_brokers/alert_broker_ztf.py
@@ -559,6 +559,8 @@ def update_fp_hists(self, alert, formatted_fp_hists):
             self.verbose > 1,
         ):
             update_pipeline = [
+                # 0. match the document
+                {"$match": {"_id": alert["objectId"]}},
                 # 1. concat the new fp_hists with the existing ones
                 {
                     "$project": {
@@ -617,6 +619,7 @@ def update_fp_hists(self, alert, formatted_fp_hists):
                     self.mongo.db[self.collection_alerts_aux]
                     .aggregate(
                         update_pipeline,
+                        allowDiskUse=True,
                     )
                     .next()
                     .get("fp_hists", [])