If you run locally, you should randomly sample a partial data set. For example: df = df.sample(frac=0.10, random_state=99)