def METHOD_NAME(self):
    with self.assertRaisesRegex(
        NotImplementedError,
        "DTensor does not support row-wise sharded embedding operation yet!",
    ):
        self._run_embedding_op_test(0, [5, 12], 16, 22)