fix rmse util

ziatdinovmax · Oct 27, 2024 · d13ab99 · d13ab99
1 parent da11286
commit d13ab99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neurobayes/utils/utils.py b/neurobayes/utils/utils.py
@@ -92,8 +92,8 @@ def rmse(y_pred: jnp.ndarray, y_true: jnp.ndarray) -> jnp.ndarray:
     """
     Calculates the root mean squared error between true and predicted values.
     """
-    mse = mse(y_pred, y_true)
-    return jnp.sqrt(mse)
+    mse_ = mse(y_pred, y_true)
+    return jnp.sqrt(mse_)
 
 
 def mae(y_pred: jnp.ndarray, y_true: jnp.ndarray) -> jnp.ndarray: