A Note on "Regularity lemma for distal structures"

In a recent paper, Chernikov and Starchenko prove that graphs defined in distal theories have strong regularity properties, generalizing previous results about graphs defined by semi-algebraic relations. We give a shorter, purely model-theoretic proof of this fact.