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):
846
846
class TestType :
847
847
samplers = (Metropolis , Slice , HamiltonianMC , NUTS )
848
848
849
- @pytensor .config .change_flags ({ " floatX" : " float64" , " warn_float64" : " ignore"} )
849
+ @pytensor .config .change_flags (floatX = " float64" , warn_float64 = " ignore" )
850
850
def test_float64 (self ):
851
851
with pm .Model () as model :
852
852
x = pm .Normal ("x" , initval = np .array (1.0 , dtype = "float64" ))
@@ -861,7 +861,7 @@ def test_float64(self):
861
861
warnings .filterwarnings ("ignore" , ".*number of samples.*" , UserWarning )
862
862
pm .sample (draws = 10 , tune = 10 , chains = 1 , step = sampler ())
863
863
864
- @pytensor .config .change_flags ({ " floatX" : " float32" , " warn_float64" : " warn"} )
864
+ @pytensor .config .change_flags (floatX = " float32" , warn_float64 = " warn" )
865
865
def test_float32 (self ):
866
866
with pm .Model () as model :
867
867
x = pm .Normal ("x" , initval = np .array (1.0 , dtype = "float32" ))
You can’t perform that action at this time.
0 commit comments