fixed retracing in build_quadratic_1d all tests passed

Dipolar-Quantum-Gases · Sep 13, 2022 · 7011764 · 7011764
1 parent f767afd
commit 7011764
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/jaxfit/__pycache__/common_jax.cpython-39.pyc b/jaxfit/__pycache__/common_jax.cpython-39.pyc
diff --git a/jaxfit/__pycache__/trf.cpython-39.pyc b/jaxfit/__pycache__/trf.cpython-39.pyc
diff --git a/jaxfit/common_jax.py b/jaxfit/common_jax.py
@@ -56,9  56,10 @@ def build_quadratic_1d(self, J, g, s, diag=None, s0=None):
         b = np.dot(g, s)
 
         if s0 is not None:
-            s0_jnp = jnp.array(s)
             s0_jnp = jnp.array(s0)
             u_jnp = self.js0_dot(J, s0_jnp)
             u = u_jnp.copy()
 
             b  = np.dot(u, v)
             c = 0.5 * np.dot(u, u)   np.dot(g, s0)
             if diag is not None:
@@ -67,8  68,8 @@ def build_quadratic_1d(self, J, g, s, diag=None, s0=None):
             return a, b, c
         else:
             return a, b
-    
-    
 
 
     def compute_jac_scale(self, J, scale_inv_old=None):
         """Compute variables scale based on the Jacobian matrix."""