From 3d3c9441a1b02647801cf102a184e34d69115bac Mon Sep 17 00:00:00 2001
From: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Date: Wed, 26 Jun 2024 22:02:07 +0100
Subject: [PATCH] missed some instances during rnenaming

---
 src/mcmc/abstractmcmc.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/mcmc/abstractmcmc.jl b/src/mcmc/abstractmcmc.jl
index 9d2ebb9ea..539074fbd 100644
--- a/src/mcmc/abstractmcmc.jl
+++ b/src/mcmc/abstractmcmc.jl
@@ -48,12 +48,12 @@ end
 function varinfo_from_logdensityfn(f::LogDensityProblemsAD.ADGradientWrapper)
     return varinfo_from_logdensityfn(parent(f))
 end
-_varinfo(f::DynamicPPL.LogDensityFunction) = f.varinfo
+varinfo_from_logdensityfn(f::DynamicPPL.LogDensityFunction) = f.varinfo
 
 function varinfo(state::TuringState)
     θ = getparams(getmodel(state.logdensity), state.state)
     # TODO: Do we need to link here first?
-    return DynamicPPL.unflatten(_varinfo(state.logdensity), θ)
+    return DynamicPPL.unflatten(varinfo_from_logdensityfn(state.logdensity), θ)
 end
 
 # NOTE: Only thing that depends on the underlying sampler.