s/tf.contrib.eager.py_func/tf.py_function/
s/tf.contrib.eager.py_func/tf.py_function/

As the latter symbol exists in both TF 1.x and 2.x.

PiperOrigin-RevId: 229779554
Change-Id: I004fc448dba11ba5b0f01a50758540df08babf1b

tensorflow/agents,tensorflow/agents