code: def forward(self, graph):
      graph.node_feat["atom_feat"] = self.atom_embedding(
          graph.node_feat["feat"]
      )
      if self.x_as_list:
          graph.node_feat["atom_feat"] = [graph.node_feat["atom_feat"]]
      if hasattr(self, "edge_embedding"):
          graph.edge_feat["bond_feat"] = self.edge_embedding(
              graph.edge_feat["feat"]
          )
      return graph
  
imports: import paddle
  import paddle.nn
  import paddle.nn.functional
  import pgl
  import pgl.nn
  from pgl.nn import functional
  from mol_encoder import AtomEncoder, BondEncoder
label_desc: Use image node_id 2 for calling the OGBMolEmbedding obj's underlying member method code with example usage: obj.forward(graph) and returns: graph