pyspark(SparkのPython)でRDDを圧縮ファイルとして保存する方法

scalaやjavaからRDDを操作するときは下記のように
1. 圧縮コーデッククラスをimportする
2. saveAsTextFileの第２引数にコーデッククラスのクラス・インスタンスを指定する
のようにするとRDDを圧縮ファイルとして保存できるのですが、

import org.apache.hadoop.io.compress.GzipCodec
sc.textFile( "/path/input/file" )
  .saveAsTextFile( "/path/output/file", classOf[GzipCodec] )

pythonではどうするんだろうと思って調べたら下記のようでした。

codec = "org.apache.hadoop.io.compress.GzipCodec"
sc.textFile( "/path/input/file" ) \
  .saveAsTextFile( "/path/output/file", codec )

参考: https://spark.apache.org/docs/latest/api/python/pyspark.html