def action_from_trinary_to_env(action) -> int:
    """
    Maps trinary model output to int action understandable by env
    """
    assert action in (0, 1, 2), f'Wrong action: {action}'
    return {
        0: 0,
        1: 2,
        2: 5
    }[action]