Update

awslabs · Dec 14, 2023 · e9027ef · e9027ef
1 parent 4b015be
commit e9027ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/graphstorm/gconstruct/transform.py b/python/graphstorm/gconstruct/transform.py
@@ -1029,7 +1029,7 @@ def call(self, feats):
                 # raw_ids is a numpy array
                 # handle empty strings in raw_ids
                 raw_ids = raw_ids.astype(str)
-                nid_idx = raw_ids != ""
+                nid_idx = np.nonzero(raw_ids != "")[0]
                 nids, _ = nid_map.map_id(raw_ids[nid_idx].astype(
                     nid_map.map_key_dtype))
                 neg_ids[i][nid_idx] = nids