SPARK-3789: Updated vertex.py, edge.py and graph.py