From 261862d778910a09c5b61edcc66958519a86815e Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Mon, 5 Jun 2023 11:02:34 +0200 Subject: [PATCH] Fix auto-naming of multi-output logprobs --- pymc/logprob/abstract.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/pymc/logprob/abstract.py b/pymc/logprob/abstract.py index c04aca200ce..01bf317eff2 100644 --- a/pymc/logprob/abstract.py +++ b/pymc/logprob/abstract.py @@ -67,10 +67,15 @@ def _logprob_helper(rv, *values, **kwargs): """Helper that calls `_logprob` dispatcher.""" logprob = _logprob(rv.owner.op, values, *rv.owner.inputs, **kwargs) - for rv in values: - if rv.name: - logprob.name = f"{rv.name}_logprob" - break + name = rv.name + if (not name) and (len(values) == 1): + name = values[0].name + if name: + if isinstance(logprob, (list, tuple)): + for i, term in enumerate(logprob): + term.name = f"{name}_logprob.{i}" + else: + logprob.name = f"{name}_logprob" return logprob