def METHOD_NAME(
    self, distribution, use_numpy, use_validation_data
):
    with self.assertRaisesRegex(
        ValueError, "not yet supported with tf.distribute.Strategy"
    ):
        self.run_correctness_test(
            distribution,
            use_numpy,
            use_validation_data,
            is_stateful_model=True,
        )