4270c3dae0b80b384c4a0026e2043345904b4fe5,deepchem/featurizers/atomic_coordinates.py,NeighborListAtomicCoordinates,_featurize,#NeighborListAtomicCoordinates#Any#,161

Before Change

    ----------
    
    N = mol.GetNumAtoms()
    coords = np.zeros((N,3))

    coords_raw = [mol.GetConformer(0).GetAtomPosition(i) for i in range(N)]
    for atom in range(N):
      coords[atom,0] = coords_raw[atom].x
      coords[atom,1] = coords_raw[atom].y
      coords[atom,2] = coords_raw[atom].z

    x_bins, y_bins, z_bins = get_cells(coords, self.neighbor_cutoff)

After Change

          if np.linalg.norm(coords[atom] - coords[neighbor_atom]) < self.neighbor_cutoff:
            neighbor_list[atom].add(neighbor_atom)
          ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// DEBUG
          all_nbrs.add(neighbor_atom)
          ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// DEBUG
          
      ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// DEBUG
      print("All neighbor-cell atoms for %d = %s" % (atom, str(all_nbrs)))

Instances