avoid creating float64 tensors when possible

ferrine · ferrine · commit d5dc9cc49c93 · 2023-06-22T08:43:00.000Z
diff --git 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):

Original file line number	Diff line number	Diff line change
`@@ -1073,11 +1073,11 @@ def symbolic_normalizing_constant(self):`
`1073`	`1073`	`for v in self.group`
`1074`	`1074`	`if isinstance(v.owner.op, MinibatchRandomVariable)`
`1075`	`1075`	`]`
`1076`		`- + [1.0] # To avoid empty max`
	`1076`	`+ + [pm.floatX(1.0)] # To avoid empty max`
`1077`	`1077`	`)`
`1078`	`1078`	`)`
`1079`	`1079`	`t = self.symbolic_single_sample(t)`
`1080`		`- return pm.floatX(t)`
	`1080`	`+ return t`
`1081`	`1081`
`1082`	`1082`	`@node_property`
`1083`	`1083`	`def symbolic_logq_not_scaled(self):`