move pyspark comment and doctest to correct location