use python3 version

kubeflow · Feb 18, 2020 · bfa9da1 · bfa9da1
1 parent 00ea3ec
commit bfa9da1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"