Skip to content

Commit

Permalink
Add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
thvasilo committed Jul 22, 2024
1 parent 6c7807d commit 3cd450e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/graphstorm/gconstruct/transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -951,6 +951,7 @@ def call(self, feats):
feats = feats[:, :self.truncate_dim]
else:
assert isinstance(feats, ExtMemArrayWrapper)
# Need to convert to in-memory array to make truncation possible
feats = feats.to_numpy()[:, :self.truncate_dim]
return {self.feat_name: feats}

Expand Down

0 comments on commit 3cd450e

Please sign in to comment.