float32 elemwise

Author: jessegrabowski

tests/tensor/test_elemwise.py

@@ -11,10 +11,9 @@
 import pytensor.scalar as ps
 import pytensor.tensor as pt
 import tests.unittest_tools as utt
-from pytensor import In, Out, grad
+from pytensor import In, Out, config, grad
 from pytensor.compile.function import function
 from pytensor.compile.mode import Mode
-from pytensor.configdefaults import config
 from pytensor.graph.basic import Apply, Variable
 from pytensor.graph.fg import FunctionGraph
 from pytensor.graph.replace import vectorize_node
@@ -1088,7 +1087,7 @@ def L_op(self, inputs, outputs, output_gradients):
     x = vector("x")
     y, _ = op(x)
     np.testing.assert_array_equal(
-        grad(y.sum(), x).eval({x: np.full((12,), np.nan)}),
-        np.ones((12,)),
+        grad(y.sum(), x).eval({x: np.full((12,), np.nan, dtype=config.floatX)}),
+        np.ones((12,), dtype=config.floatX),
         strict=True,
     )