Skip to content

GitLab

Commit cf58c635 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Set tfds_download to True to fix sentence_prediction_dataloader_test.py in oss 

PiperOrigin-RevId: 356906879
parent 13c8470b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/nlp/data/sentence_prediction_dataloader_test.py
View file @ cf58c635
...@@ -164,6 +164,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase, ...@@ -164,6 +164,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase,
input_path='' if use_tfds else tf_record_path, input_path='' if use_tfds else tf_record_path,
tfds_name='glue/mrpc' if use_tfds else '', tfds_name='glue/mrpc' if use_tfds else '',
tfds_split='train' if use_tfds else '', tfds_split='train' if use_tfds else '',
tfds_download=True,
text_fields=text_fields, text_fields=text_fields,
global_batch_size=batch_size, global_batch_size=batch_size,
seq_length=seq_length, seq_length=seq_length,
...@@ -195,6 +196,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase, ...@@ -195,6 +196,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase,
input_path='' if use_tfds else tf_record_path, input_path='' if use_tfds else tf_record_path,
tfds_name='glue/mrpc' if use_tfds else '', tfds_name='glue/mrpc' if use_tfds else '',
tfds_split='train' if use_tfds else '', tfds_split='train' if use_tfds else '',
tfds_download=True,
text_fields=text_fields, text_fields=text_fields,
global_batch_size=batch_size, global_batch_size=batch_size,
seq_length=seq_length, seq_length=seq_length,
...@@ -228,6 +230,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase, ...@@ -228,6 +230,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase,
input_path='' if use_tfds else tf_record_path, input_path='' if use_tfds else tf_record_path,
tfds_name='glue/mrpc' if use_tfds else '', tfds_name='glue/mrpc' if use_tfds else '',
tfds_split='train' if use_tfds else '', tfds_split='train' if use_tfds else '',
tfds_download=True,
text_fields=text_fields, text_fields=text_fields,
global_batch_size=batch_size, global_batch_size=batch_size,
seq_length=seq_length, seq_length=seq_length,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment