with self.assertRaisesRegex(
        TypeError,
        r"Expected output types \(tf.int64, tf.float64\) but got dataset with "
        r"output types \(tf.int32, tf.float32\)."):
      iterator.make_initializer(