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 feature specification or tf.TypeSpecs.

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