d7a7dd6fe7106f6501e9995a92153f65238bbb47,tensorflow_transform/analyzers.py,QuantilesCombiner,_make_get_buckets_callable,#QuantilesCombiner#Any#Any#,1295

Before Change

  def _make_get_buckets_callable(self, qaccumulator, stamp_token):
    final_summary = tf.compat.v1.placeholder(dtype=tf.string, shape=[])

    add_final_summary_op = qaccumulator.add_prebuilt_summary(
        stamp_token=stamp_token, summary=final_summary)

    // Create ops to flush the accumulator and return approximate boundaries.
    with tf.control_dependencies([add_final_summary_op]):
      flush_op = qaccumulator.flush(

After Change

      return buckets

    final_summary = tf.compat.v1.placeholder(dtype=tf.string,
                                             shape=[self._num_features])

    add_final_summary_op = [
        qaccumulators[summary_ix].add_prebuilt_summary(
            stamp_token=stamp_token, summary=final_summary[summary_ix])
        for summary_ix in range(self._num_features)
    ]

    // Create ops to flush the accumulator and return approximate boundaries.
    with tf.control_dependencies(add_final_summary_op):
      flush_op = [qacc.flush(stamp_token=stamp_token,

Instances