remove unecessary casting

pyro-ppl · Dec 21, 2024 · a7984ce · a7984ce
1 parent 6dd9396
commit a7984ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/numpyro/contrib/hsgp/approximation.py b/numpyro/contrib/hsgp/approximation.py
@@ -25,14 +25,14 @@ def _non_centered_approximation(phi: Array, spd: Array, m: int) -> Array:
     with numpyro.plate("basis", m):
         beta = numpyro.sample("beta", dist.Normal(loc=0.0, scale=1.0))
 
-    return jnp.asarray(phi @ (spd * beta))
+    return phi @ (spd * beta)
 
 
 def _centered_approximation(phi: Array, spd: Array, m: int) -> Array:
     with numpyro.plate("basis", m):
         beta = numpyro.sample("beta", dist.Normal(loc=0.0, scale=spd))
 
-    return jnp.asarray(phi @ beta)
+    return phi @ beta
 
 
 def linear_approximation(