code: def atom_features(datapoint: RDKitMol) -> np.ndarray:
      for atom in datapoint.GetAtoms():
          if atom.GetSymbol() not in ATOM_TYPES.keys():
              raise Exception(
                  "We only support 'H', 'C', 'N', 'O' and 'F' at this point for MXMNet Model"
              )
  
      return np.asarray(
          [
              [ATOM_TYPES[atom.GetSymbol()]]
              for atom in datapoint.GetAtoms()
          ],
          dtype=float,
      )
  
file_path: deepchem/deepchem/feat/molecule_featurizers/mxmnet_featurizer.py
imports: from rdkit import Chem
  import numpy
  import logging
  from typing import List, Optional
  from deepchem.utils.typing import RDKitMol
  from deepchem.feat.base_classes import MolecularFeaturizer
  from deepchem.feat.graph_data import GraphData
label_desc: Use image node_id 1 for calling a global function with example usage: atom_features(datapoint) and returns: np