code: def __init__(self, dim, embed_edge=True, x_as_list=False):
      super(OGBMolEmbedding, self).__init__()
      self.atom_embedding = AtomEncoder(emb_dim=dim)
      if embed_edge:
          self.edge_embedding = BondEncoder(emb_dim=dim)
      self.x_as_list = x_as_list
  
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 1 to create a new OGBMolEmbedding object from inherited base classes: nn with example: obj = OGBMolEmbedding(dim, embed_edge, x_as_list)