f7fd8a8395cbb90973dbd3ab7b55cda80865b656,tensorflow_datasets/scripts/download_and_prepare.py,,main,#Any#,57

Before Change

  if not FLAGS.compute_stats:
    return
  // TODO(rsepassi): Get splits from info
  for split in [tfds.Split.TRAIN, tfds.Split.TEST]:
    compute_stats(builder, split)


def compute_stats(builder, split):
  Print statistics for this split.
  dataset = builder.as_dataset(split=split)
  if FLAGS.debug:

After Change

  if FLAGS.debug:
    dataset = builder.as_dataset(split=tfds.Split.TRAIN)
    iterator = tf.contrib.eager.Iterator(dataset)
    pdb.set_trace()
    del iterator
    return

Instances