Add tf.float32 to unittest args

tfboyd · tfboyd · commit 5c0c749b5e4f · 2018-09-05T16:17:21.000-07:00
diff --git a/official/resnet/cifar10_test.py b/official/resnet/cifar10_test.py
@@ -61,7 +61,7 @@ def test_dataset_input_fn(self):
     fake_dataset = tf.data.FixedLengthRecordDataset(
         filename, cifar10_main._RECORD_BYTES)  # pylint: disable=protected-access
     fake_dataset = fake_dataset.map(
-        lambda val: cifar10_main.parse_record(val, False))
+        lambda val: cifar10_main.parse_record(val, False, tf.float32))
     image, label = fake_dataset.make_one_shot_iterator().get_next()
 
     self.assertAllEqual(label.shape, ())
