diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/estimator_test.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/estimator_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py b/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py index b4c8cec514..da23873225 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py +++ b/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py @@ -763,7 +763,7 @@ class EstimatorTest(test.TestCase): def __init__(self): self.run_count = 0 - def before_run(self, run_context): + def after_run(self, run_context, run_values): self.run_count += 1 est = learn.Estimator(model_fn=linear_model_fn) |