lint fix

Author: priyakasimbeg

algoperf/workloads/ogbg/ogbg_jax/models.py

@@ -93,4 +93,4 @@ def __call__(self, graph, train, dropout_rate=DROPOUT_RATE):
     decoder = jraph.GraphMapFeatures(embed_global_fn=nn.Dense(self.num_outputs))
     graph = decoder(graph)
 
-    return graph.globals
+    return graph.globals