204b4cad88ff526006e2fa2c6604c35902cf5aef,tensorflow_transform/beam/impl.py,AnalyzeDataset,_extract_input_pvalues,#AnalyzeDataset#Any#,717

Before Change

  def _extract_input_pvalues(self, dataset):
    data, metadata = dataset
    pvalues = [data] + getattr(metadata, "pcollections", {}).values()
    return dataset, pvalues

  def expand(self, dataset):
    Analyze the dataset.

After Change

  def _extract_input_pvalues(self, dataset):
    data, metadata = dataset
    pvalues = [data]
    if isinstance(metadata, beam_metadata_io.BeamDatasetMetadata):
      pvalues.append(metadata.deferred_metadata)
    return dataset, pvalues

  def expand(self, dataset):
    Analyze the dataset.

Instances