tf.train.experimental.PythonState