From 56e852320678a33477c79e0ebceafdd2d5784afc Mon Sep 17 00:00:00 2001 From: zeyus Date: Sun, 11 Aug 2024 17:06:04 +0200 Subject: [PATCH] Fix handling of unconventional variable names --- ext/ParetoSmoothDynamicPPLExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ext/ParetoSmoothDynamicPPLExt.jl b/ext/ParetoSmoothDynamicPPLExt.jl index 650dec5..6324ce6 100644 --- a/ext/ParetoSmoothDynamicPPLExt.jl +++ b/ext/ParetoSmoothDynamicPPLExt.jl @@ -37,7 +37,7 @@ function ParetoSmooth.pointwise_log_likelihoods(model::DynamicPPL.Model, chains: # Size of array (n_steps, n_chains) using first parameter dims = size(last(first(log_like_dict))) # parse "var[i]" -> i - ind_from_string(x) = parse(Int, split(split(x, "[")[2], "]")[1]) + ind_from_string(x) = parse(Int, rsplit(rsplit(x, "[", limit=2)[2], "]", limit=2)[1]) # collect variable names sorted_keys = sort(collect(keys(log_like_dict)); by=ind_from_string) # Convert from dictionary to 3d array