Skip to content

Commit

Permalink
Remove debug print statements from CoulombPotential class
Browse files Browse the repository at this point in the history
  • Loading branch information
E-Rum committed Feb 9, 2025
1 parent e8bd8f6 commit c63e026
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/torchpme/potentials/coulomb.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,8 +77,6 @@ def lr_from_k_sq(self, k_sq: torch.Tensor) -> torch.Tensor:
# https://github.com/jax-ml/jax/issues/1052
# https://github.com/tensorflow/probability/blob/main/discussion/where-nan.pdf
masked = torch.where(k_sq == 0, 1.0, k_sq)
print(self.smearing.device)
print(k_sq.device)
return torch.where(
k_sq == 0,
0.0,
Expand Down

0 comments on commit c63e026

Please sign in to comment.