from pyspark.sql.functions import col, desc

@time_decorator
def no_salting(data):
  # Perform aggregation
  agg_data = data.groupBy("customerID").agg({"totalAmt": "sum"}).sort(desc("sum(totalAmt)"))
  return agg_data

display(no_salting(df))