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 @@ -677,7 +677,7 @@ def test_diag_blockdiag_rewrite():
677
677
assert not any (isinstance (node .op , BlockDiagonal ) for node in nodes )
678
678
679
679
# Value Test
680
- sub_matrices_test = np .random .rand (n_matrices , * matrix_size )
680
+ sub_matrices_test = np .random .rand (n_matrices , * matrix_size ). astype ( config . floatX )
681
681
bd_output_test = scipy .linalg .block_diag (
682
682
* [sub_matrices_test [i ] for i in range (n_matrices )]
683
683
)
@@ -704,7 +704,7 @@ def test_det_blockdiag_rewrite():
704
704
assert not any (isinstance (node .op , BlockDiagonal ) for node in nodes )
705
705
706
706
# Value Test
707
- sub_matrices_test = np .random .rand (n_matrices , * matrix_size )
707
+ sub_matrices_test = np .random .rand (n_matrices , * matrix_size ). astype ( config . floatX )
708
708
bd_output_test = scipy .linalg .block_diag (
709
709
* [sub_matrices_test [i ] for i in range (n_matrices )]
710
710
)
@@ -733,7 +733,7 @@ def test_slogdet_blockdiag_rewrite():
733
733
assert not any (isinstance (node .op , BlockDiagonal ) for node in nodes )
734
734
735
735
# Value Test
736
- sub_matrices_test = np .random .rand (n_matrices , * matrix_size )
736
+ sub_matrices_test = np .random .rand (n_matrices , * matrix_size ). astype ( config . floatX )
737
737
bd_output_test = scipy .linalg .block_diag (
738
738
* [sub_matrices_test [i ] for i in range (n_matrices )]
739
739
)
You can’t perform that action at this time.
0 commit comments