Merge pull request #11 from alercebroker/hotfix/produce-key

hotfix: produce with aid key
alercebroker · May 13, 2022 · 16b2f0e · 16b2f0e
2 parents f6ed070 + 68055fb
commit 16b2f0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sorting_hat_step/step.py b/sorting_hat_step/step.py
@@ -64,7 +64,7 @@ def produce(self, alerts: pd.DataFrame) -> None:
                 if isinstance(alert["extra_fields"][k], list):
                     alert["extra_fields"][k] = pickle.dumps(alert["extra_fields"][k])
             # produce alert content with key of candid
-            self.producer.produce(alert, key=str(alert["candid"]))
+            self.producer.produce(alert, key=str(alert["aid"]))
             n_messages += 1
         self.logger.info(f"{n_messages} messages Produced")