Remove unnecessary set()
Moved PySpark import to avoid dependency error