diff --git a/visualize/distributions.py b/visualize/distributions.py index 0180b28..4eec298 100644 --- a/visualize/distributions.py +++ b/visualize/distributions.py @@ -12,7 +12,7 @@ def distribution_df(dataframe, label): name = dataframe.columns[0] - dataframe = dataframe.limit(300) + dataframe = dataframe try: while True: dataframe = dataframe.select(explode(col(name)).alias(name))