Skip to content

Commit 04b6ac7

Browse files
authored
add decorators (#2096)
1 parent 3a62aec commit 04b6ac7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pymc3/variational/opvi.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,7 @@ def add_obj_updates(self, updates, obj_n_mc=None, obj_optimizer=adam, more_obj_p
174174
updates.loss = obj_target
175175

176176
@memoize
177+
@change_flags(compute_test_value='off')
177178
def step_function(self, obj_n_mc=None, tf_n_mc=None,
178179
obj_optimizer=adam, test_optimizer=adam,
179180
more_obj_params=None, more_tf_params=None,
@@ -232,6 +233,7 @@ def step_function(self, obj_n_mc=None, tf_n_mc=None,
232233
return step_fn
233234

234235
@memoize
236+
@change_flags(compute_test_value='off')
235237
def score_function(self, sc_n_mc=None, more_replacements=None, fn_kwargs=None): # pragma: no cover
236238
"""
237239
Compiles scoring function that operates which takes no inputs and returns Loss

0 commit comments

Comments
 (0)