class_docstr: Only encode the first dimension of atom integer features.
  This feature encodes just the atom type
  
  Parameters:
  num_classes: Not used!
class_docstr_tok: ["Only","encode","the","first","dimension","of","atom","integer","features",".","This","feature","encodes","just","the","atom","type","Parameters",":","num_classes",":","Not","used","!"]
code: def forward(self, batch):
      batch.x = self.atom_type_embedding(batch.x[:, 0])
  
      return batch
  
file_path: GraphGym/graphgym/models/feature_encoder_pyg.py
imports: import torch
  from ogb.utils.features import get_atom_feature_dims, get_bond_feature_dims
  import graphgym.register
label_desc: Use image node_id 2 for calling the SingleAtomEncoder obj's underlying member method code with example usage: obj.forward(batch) and returns: batch