Use resource_loader to access in-tree data.
PiperOrigin-RevId: 292242396 Change-Id: Id42aaf57d8de0b1605b36881f33a3b86216de1d0
This commit is contained in:
parent
f3117e8ec1
commit
0357d7c0b4
@ -28,10 +28,11 @@ import six
|
||||
|
||||
from tensorflow.python.framework import errors_impl
|
||||
from tensorflow.python.lib.io import tf_record
|
||||
from tensorflow.python.platform import resource_loader
|
||||
from tensorflow.python.platform import test
|
||||
from tensorflow.python.util import compat
|
||||
|
||||
prefix_path = "third_party/tensorflow/core/lib"
|
||||
prefix_path = resource_loader.get_path_to_datafile("../../../core/lib")
|
||||
|
||||
TFRecordCompressionType = tf_record.TFRecordCompressionType
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user