async def corofunc(**local_args):
            nonlocal called
            assert local_args == args
            called += 1
def get_config(self):
    config = super().get_config()
    config['masked_tokens'] = tuple(self._masked_tokens)
    return config