fix test to initialize broadcast properly

ferrine · ferrine · commit b5a5024318de · 2022-12-28T15:01:03.000+03:00
diff --git a/tests/tensor/test_elemwise.py b/tests/tensor/test_elemwise.py
@@ -287,11 +287,11 @@ def with_linker_inplace(self, linker, op, type, rand_val):
                     # type shape provided by PyTensor was broadcastable/non-broadcastable
                     x_type = type(
                         pytensor.config.floatX,
-                        broadcastable=tuple(s if s == 1 else None for s in xsh),
+                        broadcastable=tuple(s == 1 for s in xsh),
                     )
                     y_type = type(
                         pytensor.config.floatX,
-                        broadcastable=tuple(s if s == 1 else None for s in ysh),
+                        broadcastable=tuple(s == 1 for s in xsh),
                     )
                 else:
                     x_type = type(pytensor.config.floatX, shape=[None for _ in xsh])

Original file line number	Diff line number	Diff line change
`@@ -287,11 +287,11 @@ def with_linker_inplace(self, linker, op, type, rand_val):`
`287`	`287`	`# type shape provided by PyTensor was broadcastable/non-broadcastable`
`288`	`288`	`x_type = type(`
`289`	`289`	`pytensor.config.floatX,`
`290`		`- broadcastable=tuple(s if s == 1 else None for s in xsh),`
	`290`	`+ broadcastable=tuple(s == 1 for s in xsh),`
`291`	`291`	`)`
`292`	`292`	`y_type = type(`
`293`	`293`	`pytensor.config.floatX,`
`294`		`- broadcastable=tuple(s if s == 1 else None for s in ysh),`
	`294`	`+ broadcastable=tuple(s == 1 for s in xsh),`
`295`	`295`	`)`
`296`	`296`	`else:`
`297`	`297`	`x_type = type(pytensor.config.floatX, shape=[None for _ in xsh])`