Random walk over graph using ptyhon network x

I have built a TQ graph (term query graph) with some user queries and I am trying to implement a random walk with restart to generate query suggestions but I’m running into some issues.

I’m using python/network x for the implementation.

reference:
https://medium.com/pinterest-engineering/evolving-search-recommendations-on-pinterest-136e26e0468a

Deliverables
working solution capable of generating query recommendations

module to construct graph.