fix: handling duplicated oids

alercebroker · Nov 3, 2023 · 873ce8b · 873ce8b
1 parent aef912f
commit 873ce8b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/metadata_step/metadata_step/step.py b/metadata_step/metadata_step/step.py
@@ -30,7 +30,9 @@ def _write_metadata_into_db(self, result: List[Dict]):
 
     # Output format: [{oid: OID, ss: SS_DATA, ...}]
     def execute(self, messages: List[Dict]):
-        oids = list(set([message["oid"] for message in messages]))
+        unique = {message["oid"]: message for message in messages}
+        oids = list(unique.keys())
+        messages = list(unique.values())
         catalogs = {"ps1": {}, "gaia": {}}
         with self.db.session() as session:
             catalogs["ps1"] = get_ps1_catalog(session, oids)