From d5dc9cc49c93ceddb9c91c011cf35d05278da13f Mon Sep 17 00:00:00 2001 From: Max Kochurov Date: Thu, 22 Jun 2023 08:43:00 +0000 Subject: [PATCH] avoid creating float64 tensors when possible --- pymc/variational/opvi.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pymc/variational/opvi.py b/pymc/variational/opvi.py index 7280105d03..d5b20c5390 100644 --- a/pymc/variational/opvi.py +++ b/pymc/variational/opvi.py @@ -1073,11 +1073,11 @@ def symbolic_normalizing_constant(self): for v in self.group if isinstance(v.owner.op, MinibatchRandomVariable) ] - + [1.0] # To avoid empty max + + [pm.floatX(1.0)] # To avoid empty max ) ) t = self.symbolic_single_sample(t) - return pm.floatX(t) + return t @node_property def symbolic_logq_not_scaled(self):