Attempt to fix missing import in tf-nightly:
Attempt to fix missing import in tf-nightly:

Fixes: https://travis-ci.org/tensorflow/adanet/jobs/527123258
PiperOrigin-RevId: 246351702

tensorflow/adanet,tensorflow/adanet