File tree 2 files changed +8
-3
lines changed
2 files changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -190,6 +190,8 @@ def walk_model(
190
190
expand_fn
191
191
A function that returns the next variable(s) to be traversed.
192
192
"""
193
+ warnings .warn ("walk_model will be removed in a future relase of PyMC" , FutureWarning )
194
+
193
195
if stop_at_vars is None :
194
196
stop_at_vars = set ()
195
197
Original file line number Diff line number Diff line change @@ -281,21 +281,24 @@ def test_walk_model():
281
281
282
282
test_graph = pt .exp (e + 1 )
283
283
284
- res = list (walk_model ((test_graph ,)))
284
+ with pytest .warns (FutureWarning ):
285
+ res = list (walk_model ((test_graph ,)))
285
286
assert a in res
286
287
assert b in res
287
288
assert c in res
288
289
assert d in res
289
290
assert e in res
290
291
291
- res = list (walk_model ((test_graph ,), stop_at_vars = {c }))
292
+ with pytest .warns (FutureWarning ):
293
+ res = list (walk_model ((test_graph ,), stop_at_vars = {c }))
292
294
assert a not in res
293
295
assert b not in res
294
296
assert c in res
295
297
assert d in res
296
298
assert e in res
297
299
298
- res = list (walk_model ((test_graph ,), stop_at_vars = {b }))
300
+ with pytest .warns (FutureWarning ):
301
+ res = list (walk_model ((test_graph ,), stop_at_vars = {b }))
299
302
assert a not in res
300
303
assert b in res
301
304
assert c in res
You can’t perform that action at this time.
0 commit comments