Watch-your-step model with StellarGraph is not working on a GPU

Question

I am trying to train a large graph-embedding using WatchYourStep algorithm using StellarGraph.

For some reason, the model is only trained on a CPU and not uti