ed9397c72665d7cca7961a93e3311f8dca5ed692,tensorflow_datasets/core/dataset_builder_test.py,DatasetBuilderReadTest,test_supervised_keys,#DatasetBuilderReadTest#,265

Before Change

  @tf.contrib.eager.run_test_in_graph_and_eager_modes()
  def test_supervised_keys(self):
    x, _ = self.builder.as_numpy(
        split=splits_lib.Split.TRAIN, as_supervised=True, batch_size=-1)
    self.assertEqual(x.shape[0], 20)


if __name__ == "__main__":

After Change

  @tf.contrib.eager.run_test_in_graph_and_eager_modes()
  def test_supervised_keys(self):
    x, _ = dataset_utils.dataset_as_numpy(self.builder.as_dataset(
        split=splits_lib.Split.TRAIN, as_supervised=True, batch_size=-1))
    self.assertEqual(x.shape[0], 20)


if __name__ == "__main__":

Instances