def METHOD_NAME(self, action_mask):
    action_mask_min = action_mask.min()
    action_mask_max = action_mask.max()
    assert (action_mask_min == 0 or action_mask_min == 1)
    assert (action_mask_max == 0 or action_mask_max == 1)
    assert (not any(action_mask.sum(dim=1) == 0))