Подскажите, пожалуйста, как правильно писать на pyspark чтоб не было

вот такого неочень осмысленного полотна?
```python
if is_target_partitioned:
anonymized_source_df \
.coalesce(20) \
.write \
.format('delta') \
.mode("overwrite") \
.partitionBy("load_date") \
.save(target_s3_path)
else:
anonymized_source_df \
.coalesce(20) \
.write \
.format('delta') \
.mode("overwrite") \
.save(target_s3_path)
```
из-за одного условия (партиций) нужно дублировать код
очень похоже на паттерн Builder, но в данном случае не понятно возвращается ли какой-то обьект обратно и если да, то какой.
ссылки на гугл тоже приветствуются)

тогда уж вот так вот, наверное?) бо по-другому у меня лыжи не едут)) https://pastebin.com/ehWdqtVi