Update tests/test_potentials.py

Co-authored-by: Philip Loche <philip.loche@posteo.de>
lab-cosmo · Jan 8, 2025 · 7755255 · 7755255
1 parent 49a34c0
commit 7755255
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_potentials.py b/tests/test_potentials.py
@@ -587,7 +587,7 @@ def test_potential_device(potential_class, device):
     exponent = 1.0
     dtype = torch.float64
 
-    if potential_class == InversePowerLawPotential:
+    if potential_class is InversePowerLawPotential:
         potential = potential_class(
             exponent=exponent, smearing=smearing, dtype=dtype, device=device
         )