Fix plot labels (#353)

TuringLang · Feb 22, 2022 · c26e5cf · c26e5cf · cpfiffer · Feb 22, 2022
1 parent 53bf939
commit c26e5cf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -3,7 +3,7 @@ uuid = "c7f686f2-ff18-58e9-bc7b-31028e88f75d"
 keywords = ["markov chain monte carlo", "probablistic programming"]
 license = "MIT"
 desc = "Chain types and utility functions for MCMC simulations."
-version = "5.0.3"
+version = "5.0.4"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"

diff --git a/src/plot.jl b/src/plot.jl
@@ -39,11 +39,11 @@ const supportedplots = push!(collect(keys(translationdict)), :mixeddensity, :cor
 
     if colordim == :parameter
         title --> "Chain $(MCMCChains.chains(c)[i])"
-        label --> string.(names(c))
+        label --> permutedims(map(string, names(c)))
         val = c.value[:, :, i]
     elseif colordim == :chain
         title --> string(names(c)[i])
-        label --> map(x -> "Chain $x", MCMCChains.chains(c))
+        label --> permutedims(map(x -> "Chain $x", MCMCChains.chains(c)))
         val = c.value[:, i, :]
     else
         throw(ArgumentError("`colordim` must be one of `:chain` or `:parameter`"))

diff --git a/test/plot_test.jl b/test/plot_test.jl
@@ -70,4 +70,4 @@ Logging.disable_logging(Logging.Warn)
 end
 
 # Reset log level.
-Logging.disable_logging(level)
+Logging.disable_logging(level)