Skip to content

Commit

Permalink
Fix linter error
Browse files Browse the repository at this point in the history
  • Loading branch information
henryw7 committed Dec 19, 2024
1 parent baf2709 commit 3d74a26
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 7 deletions.
2 changes: 1 addition & 1 deletion gpu4pyscf/gto/int3c1e.py
Original file line number Diff line number Diff line change
Expand Up @@ -453,7 +453,7 @@ def get_int3c1e_density_contracted(mol, grids, charge_exponents, dm, intopt):
return int3c_density_contracted

def int1e_grids(mol, grids, charge_exponents=None, dm=None, charges=None, direct_scf_tol=1e-13, intopt=None):
'''
r'''
This function computes
$$\left(\mu \middle| \frac{1}{|\vec{r} - \vec{C}|} \middle| \nu\right)$$
Expand Down
7 changes: 4 additions & 3 deletions gpu4pyscf/gto/int3c1e_ip.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,7 @@ def get_int3c1e_ip2_charge_and_density_contracted(mol, grids, charge_exponents,
return int3c_ip2

def int1e_grids_ip1(mol, grids, charge_exponents=None, dm=None, charges=None, direct_scf_tol=1e-13, intopt=None):
'''
r'''
This function computes
$$\left(\frac{\partial}{\partial \vec{A}} \mu \middle| \frac{1}{|\vec{r} - \vec{C}|} \middle| \nu\right)$$
where $\mu(\vec{r})$ centers at $\vec{A}$ and $\nu(\vec{r})$ centers at $\vec{B}$.
Expand All @@ -325,7 +325,8 @@ def int1e_grids_ip1(mol, grids, charge_exponents=None, dm=None, charges=None, di
where $q_C$ is the charge centered at $\vec{C}$.
If charges is not None and dm is not None, the function computes the following contraction:
$$\sum_\nu^{n_{ao}} D_{\mu\nu} \sum_{C}^{n_{charge}} q_C \left(\frac{\partial}{\partial \vec{A}} \mu \middle| \frac{1}{|\vec{r} - \vec{C}|} \middle| \nu\right)$$
$$\sum_\nu^{n_{ao}} D_{\mu\nu} \sum_{C}^{n_{charge}} q_C
\left(\frac{\partial}{\partial \vec{A}} \mu \middle| \frac{1}{|\vec{r} - \vec{C}|} \middle| \nu\right)$$
'''
assert grids is not None

Expand All @@ -348,7 +349,7 @@ def int1e_grids_ip1(mol, grids, charge_exponents=None, dm=None, charges=None, di
return get_int3c1e_ip1_charge_contracted(mol, grids, charge_exponents, charges, intopt)

def int1e_grids_ip2(mol, grids, charge_exponents=None, dm=None, charges=None, direct_scf_tol=1e-13, intopt=None):
'''
r'''
This function computes
$$\left(\mu \middle| \frac{\partial}{\partial \vec{C}} \frac{1}{|\vec{r} - \vec{C}|} \middle| \nu\right)$$
where $\mu(\vec{r})$ centers at $\vec{A}$ and $\nu(\vec{r})$ centers at $\vec{B}$.
Expand Down
4 changes: 2 additions & 2 deletions gpu4pyscf/gto/tests/test_int1e_grids.py
Original file line number Diff line number Diff line change
Expand Up @@ -263,7 +263,7 @@ def test_int1e_grids_charge_contracted_guassian_charge_omega(self):

# Multiple inputs

def test_int1e_grids_density_contracted_sph_asymmetric(self):
def test_int1e_grids_multiple_density_contracted(self):
np.random.seed(12348)
n_dm = 4
dm = np.random.uniform(-2.0, 2.0, (n_dm, mol_sph.nao, mol_sph.nao))
Expand All @@ -274,7 +274,7 @@ def test_int1e_grids_density_contracted_sph_asymmetric(self):
assert isinstance(test_int1e_dot_D, cp.ndarray)
cp.testing.assert_allclose(ref_int1e_dot_D, test_int1e_dot_D, atol = density_contraction_threshold)

def test_int1e_grids_charge_contracted_sph(self):
def test_int1e_grids_multiple_charge_contracted(self):
np.random.seed(12348)
n_charges = 4
charges = np.random.uniform(-2.0, 2.0, (n_charges, grid_points.shape[0]))
Expand Down
1 change: 0 additions & 1 deletion gpu4pyscf/qmmm/chelpg.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,6 @@ def tau_f(R, Rcut, Roff):
r_pX = np.delete(r_pX, idx, axis=1)
gridcoords = np.delete(gridcoords, idx, axis=0)

ngrids = gridcoords.shape[0]
r_pX = cupy.array(r_pX)
r_pX_potential = 1/r_pX
potential_real = cupy.dot(cupy.array(mf.mol.atom_charges()), r_pX_potential)
Expand Down

0 comments on commit 3d74a26

Please sign in to comment.