Change

awslabs · Jul 30, 2024 · 5e7c2fc · 5e7c2fc
1 parent b217b4b
commit 5e7c2fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphstorm-processing/graphstorm_processing/config/config_conversion/gconstruct_converter.py b/graphstorm-processing/graphstorm_processing/config/config_conversion/gconstruct_converter.py
@@ -134,7 +134,7 @@ def _convert_feature(feats: list[Mapping[str, Any]]) -> list[dict]:
                         "imputer": "none",
                     }
 
-                    if gconstruct_transform_dict.get("out_dtype") in ["float32", "float64"]:
+                    if gconstruct_transform_dict.get("out_dtype") in VALID_OUTDTYPE:
                         gsp_transformation_dict["kwargs"]["out_dtype"] = gconstruct_transform_dict[
                             "out_dtype"
                         ]