From 6c7807d1ab1f3cd9dba9bb9977362296f6a284ab Mon Sep 17 00:00:00 2001 From: Theodore Vasiloudis Date: Fri, 19 Jul 2024 12:07:30 -0700 Subject: [PATCH] Update python/graphstorm/gconstruct/transform.py --- python/graphstorm/gconstruct/transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/graphstorm/gconstruct/transform.py b/python/graphstorm/gconstruct/transform.py index 0ce3cf6d0d..2a1a2f732b 100644 --- a/python/graphstorm/gconstruct/transform.py +++ b/python/graphstorm/gconstruct/transform.py @@ -1161,7 +1161,7 @@ def parse_feat_ops(confs, input_data_format=None): if 'transform' not in feat: transform = Noop( feat['feature_col'], - feat_name,\ + feat_name, out_dtype=out_dtype, truncate_dim=feat.get('truncate_dim', None) )