File tree 1 file changed +3
-3
lines changed 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -583,7 +583,7 @@ def test_diag_blockdiag_rewrite():
583
583
assert not any (isinstance (node .op , BlockDiagonal ) for node in nodes )
584
584
585
585
# Value Test
586
- sub_matrices_test = np .random .rand (n_matrices , * matrix_size )
586
+ sub_matrices_test = np .random .rand (n_matrices , * matrix_size ). astype ( config . floatX )
587
587
bd_output_test = scipy .linalg .block_diag (
588
588
* [sub_matrices_test [i ] for i in range (n_matrices )]
589
589
)
@@ -610,7 +610,7 @@ def test_det_blockdiag_rewrite():
610
610
assert not any (isinstance (node .op , BlockDiagonal ) for node in nodes )
611
611
612
612
# Value Test
613
- sub_matrices_test = np .random .rand (n_matrices , * matrix_size )
613
+ sub_matrices_test = np .random .rand (n_matrices , * matrix_size ). astype ( config . floatX )
614
614
bd_output_test = scipy .linalg .block_diag (
615
615
* [sub_matrices_test [i ] for i in range (n_matrices )]
616
616
)
@@ -639,7 +639,7 @@ def test_slogdet_blockdiag_rewrite():
639
639
assert not any (isinstance (node .op , BlockDiagonal ) for node in nodes )
640
640
641
641
# Value Test
642
- sub_matrices_test = np .random .rand (n_matrices , * matrix_size )
642
+ sub_matrices_test = np .random .rand (n_matrices , * matrix_size ). astype ( config . floatX )
643
643
bd_output_test = scipy .linalg .block_diag (
644
644
* [sub_matrices_test [i ] for i in range (n_matrices )]
645
645
)
You can’t perform that action at this time.
0 commit comments