mirror of https://github.com/kubeflow/examples.git
fix import issue in the mnist e2e testing (#531)
This commit is contained in:
parent
07b485ec8c
commit
fb0c5eb115
|
|
@ -23,7 +23,7 @@ import logging
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from kubernetes import client as k8s_client
|
from kubernetes import client as k8s_client
|
||||||
from py import test_runner #pylint: disable=no-name-in-module
|
from kubeflow.tf_operator import test_runner #pylint: disable=no-name-in-module
|
||||||
|
|
||||||
from kubeflow.testing import ks_util
|
from kubeflow.testing import ks_util
|
||||||
from kubeflow.testing import test_util
|
from kubeflow.testing import test_util
|
||||||
|
|
|
||||||
|
|
@ -28,8 +28,8 @@ import logging
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from kubernetes import client as k8s_client
|
from kubernetes import client as k8s_client
|
||||||
from py import tf_job_client #pylint: disable=no-name-in-module
|
from kubeflow.tf_operator import tf_job_client #pylint: disable=no-name-in-module
|
||||||
from py import test_runner #pylint: disable=no-name-in-module
|
from kubeflow.tf_operator import test_runner #pylint: disable=no-name-in-module
|
||||||
|
|
||||||
from kubeflow.testing import ks_util
|
from kubeflow.testing import ks_util
|
||||||
from kubeflow.testing import test_util
|
from kubeflow.testing import test_util
|
||||||
|
|
|
||||||
|
|
@ -18,13 +18,13 @@ local defaultParams = {
|
||||||
stepImage: "gcr.io/kubeflow-ci/test-worker/test-worker:v20190116-b7abb8d-e3b0c4",
|
stepImage: "gcr.io/kubeflow-ci/test-worker/test-worker:v20190116-b7abb8d-e3b0c4",
|
||||||
|
|
||||||
// Which Kubeflow cluster to use for running TFJobs on.
|
// Which Kubeflow cluster to use for running TFJobs on.
|
||||||
kfProject: "kubeflow-ci",
|
kfProject: "kubeflow-ci-deployment",
|
||||||
kfZone: "us-east1-d",
|
kfZone: "us-east1-b",
|
||||||
kfCluster: "kf-v0-4-n00",
|
kfCluster: "kf-vmaster-n00",
|
||||||
|
|
||||||
// The bucket where the model should be written
|
// The bucket where the model should be written
|
||||||
// This needs to be writable by the GCP service account in the Kubeflow cluster (not the test cluster)
|
// This needs to be writable by the GCP service account in the Kubeflow cluster (not the test cluster)
|
||||||
modelBucket: "kubeflow-ci_temp",
|
modelBucket: "kubeflow-ci-deployment_ci-temp",
|
||||||
|
|
||||||
// Whether to delete the namespace at the end.
|
// Whether to delete the namespace at the end.
|
||||||
// Leaving the namespace around can be useful for debugging.
|
// Leaving the namespace around can be useful for debugging.
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue