@@ -61,11 +61,11 @@ def test_append_invalid(self):
61
61
with pytest .raises (ValueError ):
62
62
self .strace .setup (self .draws , self .chain )
63
63
with pytest .raises (ValueError ):
64
- vars = self .sampler_vars + [{"a" : np . bool }]
64
+ vars = self .sampler_vars + [{"a" : bool }]
65
65
self .strace .setup (self .draws , self .chain , vars )
66
66
else :
67
67
with pytest .raises ((ValueError , TypeError )):
68
- self .strace .setup (self .draws , self .chain , [{"a" : np . bool }])
68
+ self .strace .setup (self .draws , self .chain , [{"a" : bool }])
69
69
70
70
def test_append (self ):
71
71
if self .sampler_vars is None :
@@ -103,7 +103,7 @@ def setup_method(self):
103
103
self .draws , self .chain = 3 , 0
104
104
105
105
def test_bad_dtype (self ):
106
- bad_vars = [{"a" : np .float64 }, {"a" : np . bool }]
106
+ bad_vars = [{"a" : np .float64 }, {"a" : bool }]
107
107
good_vars = [{"a" : np .float64 }, {"a" : np .float64 }]
108
108
with self .model :
109
109
strace = self .backend (self .name )
@@ -185,7 +185,7 @@ def setup_class(cls):
185
185
cls .expected_stats [0 ].append (stats )
186
186
cls .expected_stats [1 ].append (stats )
187
187
for key , dtype in vars .items ():
188
- if dtype == np . bool :
188
+ if dtype == bool :
189
189
stats [key ] = np .zeros (cls .draws , dtype = dtype )
190
190
else :
191
191
stats [key ] = np .arange (cls .draws , dtype = dtype )
0 commit comments