Actually, modern versions of black will retain the fluent style you prefer, though it will collapse up to the first method call on the first line within the par

Actually, modern versions of black will retain the fluent style you prefer, though it will collapse up to the first method call on the first line within the parens, so you end up with something like: foo = ( spark.read.parquet(...) .filter(...) .withColumn(...) )