fix graph_sage_stage bug

nv-morpheus · Jan 9, 2024 · 38f9e50 · 38f9e50
1 parent d1d7789
commit 38f9e50
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/gnn_fraud_detection_pipeline/stages/graph_sage_stage.py b/examples/gnn_fraud_detection_pipeline/stages/graph_sage_stage.py
@@ -95,7 +95,8 @@ def _process_message(self, message: FraudGraphMultiMessage) -> GraphSAGEMultiMes
         inductive_embedding.rename(lambda x: "ind_emb_" + str(x), axis=1, inplace=True)
 
         for col in inductive_embedding.columns.values.tolist():
-            message.set_meta(col, inductive_embedding[col])
+            # without `to_pandas`, all values in the meta become `<NA>`
+            message.set_meta(col, inductive_embedding[col].to_pandas())
 
         assert (message.mess_count == len(inductive_embedding))