Join the SIG TFX-Addons community and help make TFX even better!

tft.get_transform_input_columns

Return columns that are required inputs of TransformDataset.

preprocessing_fn A tf.transform preprocessing_fn.
specs A dict of feature name to tf.TypeSpecs. If force_tf_compat_v1 is True, this can also be feature specifications.
force_tf_compat_v1 (Optional) If True, use Tensorflow in compat.v1 mode. Defaults to True.

A list of columns that are required inputs of the transform tf.Graph defined by preprocessing_fn.