From bfa9da1a3bfdaa30353ec0dbca3c509217b8e5a0 Mon Sep 17 00:00:00 2001 From: Gabriel Wen Date: Tue, 18 Feb 2020 13:29:45 -0800 Subject: [PATCH] use python3 version --- py/kubeflow/tf_operator/simple_tfjob_tests.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/py/kubeflow/tf_operator/simple_tfjob_tests.py b/py/kubeflow/tf_operator/simple_tfjob_tests.py index 417e5cc567..078585d78f 100644 --- a/py/kubeflow/tf_operator/simple_tfjob_tests.py +++ b/py/kubeflow/tf_operator/simple_tfjob_tests.py @@ -2,7 +2,8 @@ import logging from kubeflow.testing import ks_util, test_util, util -from kubeflow.tf_operator import test_runner, tf_job_client +from kubeflow.tf_operator import test_runner +from kubeflow.tf_operator import tf_job_client_py3 as tf_job_client from kubernetes import client as k8s_client CPU_TFJOB_COMPONENT_NAME = "simple_tfjob"