orders = spark.range(0, 60, 1, numPartitions=3).select(
    (F.col("id") % 6).alias("bucket")
)