def setUpClass(cls):
super(BaseKubeflowTest, cls).setUpClass()
if ':' not in cls._BASE_CONTAINER_IMAGE:
# Generate base container image for the test if tag is not specified.
cls.container_image = '{}:{}'.format(cls._BASE_CONTAINER_IMAGE,
test_utils.random_id())
# Create a container image for use by test pipelines.
test_utils.build_and_push_docker_image(cls.container_image,
cls._REPO_BASE)
else: # Use the given image as a base image.
cls.container_image = cls._BASE_CONTAINER_IMAGE
def _prepare_base_container_image(self):
orchestration_test_utils.build_and_push_docker_image(
self._base_container_image, self._REPO_BASE)
def setUpClass(cls):
super(BaseKubeflowV2Test, cls).setUpClass()
# Create a container image for use by test pipelines.
test_utils.build_and_push_docker_image(cls._CONTAINER_IMAGE,
cls._REPO_BASE)
