How to write neo4j in python with neo4j-spark-connector

Hi Kunal,

I tried the code your provided above but got the error below

PicklingError: Could not serialize object: TypeError: Cannot serialize socket object
``
Do you have any idea how should I fix this?

Thanks,
Shuai