Use MarshalSerializer with PySpark