ab7dddff1eac1520704b86e2305ea3c49d2a3c04,tensorflow_transform/tf_utils_test.py,,,#,39

Before Change

    self._shape = shape
    self._dtype = dtype

if not hasattr(tf, "SparseTensorSpec"):
  tf.SparseTensorSpec = _SparseTensorSpec


class TFUtilsTest(test_case.TransformTestCase):

  def _assertCompositeRefEqual(self, left, right):
    Asserts that a two `tf_util._CompositeTensorRef`s are equal.

After Change

from tensorflow.python.framework import composite_tensor  // pylint: disable=g-direct-tensorflow-import

_CONSTRUCT_TABLE_PARAMETERS = [
    dict(testcase_name="_string", asset_path_input_fn=lambda x: x),
    dict(testcase_name="_string_tensor", asset_path_input_fn=tf.constant),
    dict(
        testcase_name="_saved_model_asset",

Instances