Cache underlying SchemaRDD instead of generating and caching PythonRDD