Can RGAT be implemented for a graph level multiclass-classification?

[NipunaChhabra]

I have written a graph mutli-class classification task, following the lead from the GraphBinaryClassificationTask,
and a dataset file following the lead from json_graph_property_dataset, but how do I use an rgat layer now? The PPI is an implementation of node level task, so I wanted to make sure what I'm trying to do is plausible here.