import os

import tensorflow as tf
import tensorflow_transform as tft

from tfx.components.transform import executor_test


class ExecutorV2Test(executor_test.ExecutorTest):
  # Should not rely on inherited _SOURCE_DATA_DIR for integration tests to work
  # when TFX is installed as a non-editable package.
  _SOURCE_DATA_DIR = os.path.join(
      os.path.dirname(os.path.dirname(__file__)), 'testdata')

  def _use_force_tf_compat_v1(self):
    return False


if __name__ == '__main__':
  # TODO(b/168641322): remove once TFT post-0.25.0 released and depended on.
  if tft.__version__ > '0.25.0' and tf.version.VERSION >= '2.4':
    tf.test.main()

import os

import tensorflow as tf

from tfx.components.transform import executor_test


class ExecutorV2Test(executor_test.ExecutorTest):
  # Should not rely on inherited _SOURCE_DATA_DIR for integration tests to work
  # when TFX is installed as a non-editable package.
  _SOURCE_DATA_DIR = os.path.join(
      os.path.dirname(os.path.dirname(__file__)), 'testdata')

  def _use_force_tf_compat_v1(self):
    return False


if __name__ == '__main__':
  tf.test.main()

Remove version check now that TFX 0.25 is released.
Remove version check now that TFX 0.25 is released.

PiperOrigin-RevId: 343533114

  import os
  
  import tensorflow as tf
- import tensorflow_transform as tft
  
  from tfx.components.transform import executor_test
  
  
  class ExecutorV2Test(executor_test.ExecutorTest):
    # Should not rely on inherited _SOURCE_DATA_DIR for integration tests to work
    # when TFX is installed as a non-editable package.
    _SOURCE_DATA_DIR = os.path.join(
        os.path.dirname(os.path.dirname(__file__)), 'testdata')
  
    def _use_force_tf_compat_v1(self):
      return False
  
  
  if __name__ == '__main__':
-   # TODO(b/168641322): remove once TFT post-0.25.0 released and depended on.
-   if tft.__version__ > '0.25.0' and tf.version.VERSION >= '2.4':
-     tf.test.main()
+   tf.test.main()
  
Remove version check now that TFX 0.25 is released.
## Code Before:
import os

import tensorflow as tf
import tensorflow_transform as tft

from tfx.components.transform import executor_test


class ExecutorV2Test(executor_test.ExecutorTest):
  # Should not rely on inherited _SOURCE_DATA_DIR for integration tests to work
  # when TFX is installed as a non-editable package.
  _SOURCE_DATA_DIR = os.path.join(
      os.path.dirname(os.path.dirname(__file__)), 'testdata')

  def _use_force_tf_compat_v1(self):
    return False


if __name__ == '__main__':
  # TODO(b/168641322): remove once TFT post-0.25.0 released and depended on.
  if tft.__version__ > '0.25.0' and tf.version.VERSION >= '2.4':
    tf.test.main()

## Instruction:
Remove version check now that TFX 0.25 is released.
## Code After:
import os

import tensorflow as tf

from tfx.components.transform import executor_test


class ExecutorV2Test(executor_test.ExecutorTest):
  # Should not rely on inherited _SOURCE_DATA_DIR for integration tests to work
  # when TFX is installed as a non-editable package.
  _SOURCE_DATA_DIR = os.path.join(
      os.path.dirname(os.path.dirname(__file__)), 'testdata')

  def _use_force_tf_compat_v1(self):
    return False


if __name__ == '__main__':
  tf.test.main()
