Thanks for the question Siddharth.
We started to work on a more native support for GNNs in RLlib recently.
At this point though, the best bet may be to just padding your observations to max number of nodes and features. and unpack them in your custom model.
You may find this thread interesting: