Skip to content

GitLab

Commit c556dad9 authored by Taylor Robie's avatar Taylor Robie
Browse files

delint

parent 9d42f797
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
official/recommendation/data_test.py
View file @ c556dad9
......@@ -129,8 +129,7 @@ class BaseTest(tf.test.TestCase):
def test_end_to_end(self):
params = self.make_params(train_epochs=1)
_, _, producer = data_preprocessing.instantiate_pipeline(
dataset=DATASET, data_dir=self.temp_data_dir, deterministic=False,
params=params)
dataset=DATASET, data_dir=self.temp_data_dir, params=params)
producer.start()
producer.join()
......@@ -233,8 +232,7 @@ class BaseTest(tf.test.TestCase):
train_epochs = 5
params = self.make_params(train_epochs=train_epochs)
_, _, producer = data_preprocessing.instantiate_pipeline(
dataset=DATASET, data_dir=self.temp_data_dir, deterministic=False,
params=params)
dataset=DATASET, data_dir=self.temp_data_dir, params=params)
producer.start()
......
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