File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -846,7 +846,7 @@ def test_step_vars_in_model(self):
846846class TestType :
847847 samplers = (Metropolis , Slice , HamiltonianMC , NUTS )
848848
849- @pytensor .config .change_flags ({ " floatX" : " float64" , " warn_float64" : " ignore"} )
849+ @pytensor .config .change_flags (floatX = " float64" , warn_float64 = " ignore" )
850850 def test_float64 (self ):
851851 with pm .Model () as model :
852852 x = pm .Normal ("x" , initval = np .array (1.0 , dtype = "float64" ))
@@ -861,7 +861,7 @@ def test_float64(self):
861861 warnings .filterwarnings ("ignore" , ".*number of samples.*" , UserWarning )
862862 pm .sample (draws = 10 , tune = 10 , chains = 1 , step = sampler ())
863863
864- @pytensor .config .change_flags ({ " floatX" : " float32" , " warn_float64" : " warn"} )
864+ @pytensor .config .change_flags (floatX = " float32" , warn_float64 = " warn" )
865865 def test_float32 (self ):
866866 with pm .Model () as model :
867867 x = pm .Normal ("x" , initval = np .array (1.0 , dtype = "float32" ))
You can’t perform that action at this time.
0 commit comments