Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "f7ea371ec389f22578dd9da775dc9d62486f2383"
Commit 5a0bc89a 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 b6e0e8e5
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/nlp/data/sentence_prediction_dataloader_test.py
View file @ 5a0bc89a
......@@ -164,6 +164,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase,
input_path='' if use_tfds else tf_record_path,
tfds_name='glue/mrpc' if use_tfds else '',
tfds_split='train' if use_tfds else '',
tfds_download=True,
text_fields=text_fields,
global_batch_size=batch_size,
seq_length=seq_length,
......@@ -195,6 +196,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase,
input_path='' if use_tfds else tf_record_path,
tfds_name='glue/mrpc' if use_tfds else '',
tfds_split='train' if use_tfds else '',
tfds_download=True,
text_fields=text_fields,
global_batch_size=batch_size,
seq_length=seq_length,
......@@ -228,6 +230,7 @@ class SentencePredictionTfdsDataLoaderTest(tf.test.TestCase,
input_path='' if use_tfds else tf_record_path,
tfds_name='glue/mrpc' if use_tfds else '',
tfds_split='train' if use_tfds else '',
tfds_download=True,
text_fields=text_fields,
global_batch_size=batch_size,
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