13
13
def check_trace (model , trace , n , step , start ):
14
14
#try using a trace object a few times
15
15
for i in range (2 ):
16
- trace = sample (n , step , start , trace , model = model )
16
+ trace = sample (n , step , start , trace , track_progress = False , model = model )
17
17
18
18
for (var , val ) in start .iteritems ():
19
19
@@ -23,11 +23,11 @@ def check_trace(model, trace, n, step, start):
23
23
def test_trace ():
24
24
model , start , step ,_ = simple_init ()
25
25
26
- for h in [pm .NpTrace ]:
26
+ for h in [pm .NpTrace ]:
27
27
for n in [20 , 1000 ]:
28
28
for vars in [model .vars , model .vars + [model .vars [0 ]** 2 ]]:
29
29
trace = h (vars )
30
-
30
+
31
31
32
32
yield check_trace , model , trace , n , step , start
33
33
@@ -45,8 +45,8 @@ def test_multitrace():
45
45
def check_multi_trace (model , trace , n , step , start ):
46
46
47
47
for i in range (2 ):
48
- trace = psample (n , step , start , trace , model = model )
49
-
48
+ trace = psample (n , step , start , trace , track_progress = False , model = model )
49
+
50
50
51
51
for (var , val ) in start .iteritems ():
52
52
print [len (tr .samples [var ].vals ) for tr in trace .traces ]
@@ -68,7 +68,7 @@ def test_get_point():
68
68
x = pm .NpTrace (model .vars )
69
69
x .record (p )
70
70
x .record (p2 )
71
- assert x .point (1 ) == x [1 ]
71
+ assert x .point (1 ) == x [1 ]
72
72
73
73
74
74
0 commit comments