From 663517583c96dd3bf48b449b828b18241003242c Mon Sep 17 00:00:00 2001 From: JalenCato Date: Mon, 23 Oct 2023 19:47:00 +0000 Subject: [PATCH] add concatenation --- python/graphstorm/gconstruct/transform.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/graphstorm/gconstruct/transform.py b/python/graphstorm/gconstruct/transform.py index 84c50b4cc5..9c7a5abe3a 100644 --- a/python/graphstorm/gconstruct/transform.py +++ b/python/graphstorm/gconstruct/transform.py @@ -1096,6 +1096,8 @@ def process_features(data, ops): val = val.to_numpy().reshape(-1, 1) else: val = val.reshape(-1, 1) + if key in new_data: + val = np.column_stack((new_data[key], val)) new_data[key] = val return new_data