Update example

lab-cosmo · Jan 19, 2025 · 0d335c7 · 0d335c7
1 parent b97e0b3
commit 0d335c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/10-tuning.py b/examples/10-tuning.py
@@ -19,7 +19,7 @@
 import vesin.torch as vesin
 
 import torchpme
-from torchpme.tuning.error_bounds import PMEErrorBounds
+from torchpme.tuning.pme import PMEErrorBounds
 from torchpme.tuning.tuner import TuningTimings
 
 DTYPE = torch.float64
@@ -235,7 +235,7 @@ def timed_madelung(cutoff, smearing, mesh_spacing, interpolation_nodes):
 
 # %%
 
-EB = torchpme.tuning.error_bounds.PMEErrorBounds((charges**2).sum(), cell, positions)
+EB = torchpme.tuning.pme.PMEErrorBounds((charges**2).sum(), cell, positions)
 
 # %%
 v, t = timed_madelung(cutoff=5, smearing=1, mesh_spacing=1, interpolation_nodes=4)