def get_lra_atoms(formula):
    """Finds all the LRA atoms in the formula.
    
    Args:
        formula (FNode): The pysmt formula to examine.
        
    Returns:
        set(FNode): The set of all the LRA atoms in the formula.
    
    """  
    return {a for a in formula.get_atoms() if a.is_theory_relation()}