fix bug with produce when multiple candid

alercebroker · Dec 3, 2020 · 10de022 · 10de022
1 parent 455dc19
commit 10de022
Show file tree

Hide file tree

Showing 4 changed files with 1,238 additions and 0 deletions.
diff --git a/features/step.py b/features/step.py
@@ -260,6 +260,8 @@ def produce(self, features, alert_data):
         for oid, features_oid in features.iterrows():
             features_oid.replace({np.nan: None}, inplace=True)
             candid = alert_data.loc[oid].candid
+            if isinstance(candid, pd.Series):
+                candid = candid.iloc[-1]
             features_dict = features_oid.to_dict()
             out_message = {"features": features_dict, "oid": oid, "candid": candid}
             self.producer.produce(out_message, key=oid)