def run_hook(hook,
             batch_data: dict = {'labels': [0, 1], 'predictions': [0, 1], 'masks': [1, 0]},
             epoch_data: dict = {'train': {'accuracy': 1}}):
    """
    Run hook's methods `after_batch` and `after_epoch`
    Returns modified epoch_data
    """
    hook.after_batch(stream_name='train', batch_data=batch_data)
    hook.after_epoch(epoch_id=0, epoch_data=epoch_data)
    return epoch_data