Update x0 initialization in HarmonicOscillatorPotential constructor

choderalab · Dec 29, 2023 · 409f5c7 · 409f5c7
1 parent 551ce2b
commit 409f5c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chiron/potential.py b/chiron/potential.py
@@ -272,7 +272,7 @@ def __init__(
         self,
         topology: Topology,
         k: unit.Quantity = 1.0 * unit.kilocalories_per_mole / unit.angstrom**2,
-        x0: unit.Quantity = [[0.0, 0.0, 0.0]] * unit.angstrom,
+        x0: unit.Quantity = jnp.array([[0.0, 0.0, 0.0]]) * unit.angstrom,
         U0: unit.Quantity = 0.0 * unit.kilocalories_per_mole,
     ):
         """