Update PythonPartitioner.scala