with self.assertRaisesRegex(
        TypeError,
        r"Expected output shapes compatible with .* but got dataset with "
        r"output shapes.*"):
      iterator.make_initializer(