Commit 93a652ef authored by Yong Tang's avatar Yong Tang Committed by Martin Wicke
Browse files

Add annotations support for tf.estimator.Estimator (#12276)



* Add annotations support for tf.estimator.Estimator
This fix adds annotations support for tf.estimator.Estimator
so that the following works in python 3:
```
import tensorflow as tf

def model_fn(features: dict, labels: tf.Tensor, mode: str):
    pass

estimator = tf.estimator.Estimator(model_fn)
```

This fix fixes 12249.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Update tf_inspect with getfullargspec

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 10e87011
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment