class_docstr: This class computes the featurization that corresponds to AtomicConvModel.
  
  This class computes featurizations needed for AtomicConvModel.
  Given two molecular structures, it computes a number of useful
  geometric features. In particular, for each molecule and the global
  complex, it computes a coordinates matrix of size (N_atoms, 3)
  where N_atoms is the number of atoms. It also computes a
  neighbor-list, a dictionary with N_atoms elements where
  neighbor-list[i] is a list of the atoms the i-th atom has as
  neighbors. In addition, it computes a z-matrix for the molecule
  which is an array of shape (N_atoms,) that contains the atomic
  number of that atom.
  
  Since the featurization computes these three quantities for each of
  the two molecules and the complex, a total of 9 quantities are
  returned for each complex. Note that for efficiency, fragments of
  the molecules can be provided rather than the full molecules
  themselves.
class_docstr_tok: ["This","class","computes","the","featurization","that","corresponds","to","AtomicConvModel",".","This","class","computes","featurizations","needed","for","AtomicConvModel",".","Given","two","molecular","structures",",","it","computes","a","number","of","useful","geometric","features",".","In","particular",",","for","each","molecule","and","the","global","complex",",","it","computes","a","coordinates","matrix","of","size","(","N_atoms",",","3",")","where","N_atoms","is","the","number","of","atoms",".","It","also","computes","a","neighbor-list",",","a","dictionary","with","N_atoms","elements","where","neighbor-list","[","i","]","is","a","list","of","the","atoms","the","i-th","atom","has","as","neighbors",".","In","addition",",","it","computes","a","z-matrix","for","the","molecule","which","is","an","array","of","shape","(","N_atoms",",",")","that","contains","the","atomic","number","of","that","atom",".","Since","the","featurization","computes","these","three","quantities","for","each","of","the","two","molecules","and","the","complex",",","a","total","of","9","quantities","are","returned","for","each","complex",".","Note","that","for","efficiency",",","fragments","of","the","molecules","can","be","provided","rather","than","the","full","molecules","themselves","."]
code: def featurize_mol(self, coords, mol, max_num_atoms):
      logging.info(
          "Featurizing molecule of size: %d", len(mol.GetAtoms())
      )
      neighbor_list = compute_neighbor_list(
          coords, self.neighbor_cutoff, self.max_num_neighbors, None
      )
      # pad outputs
      z = self.get_Z_matrix(mol, max_num_atoms)
      z = pad_array(z, max_num_atoms)
      coords = pad_array(coords, (max_num_atoms, 3))
      return coords, neighbor_list, z
  
imports: import logging
  import warnings
  import numpy
  from deepchem.feat.base_classes import Featurizer, ComplexFeaturizer
  from deepchem.feat.molecule_featurizers import AtomicCoordinates
  from deepchem.utils.data_utils import pad_array
  from deepchem.utils.rdkit_utils import MoleculeLoadException, get_xyz_from_mol, load_molecule, merge_molecules_xyz, merge_molecules
label_desc: Use image node_id 4 for calling the AtomicConvFeaturizer obj's underlying member method code with example usage: obj.featurize_mol(coords, mol, max_num_atoms) and returns: coords,  neighbor_list,  z