Skip to content

Commit

Permalink
Update test worker image to kubeflow-ci (kubeflow#568)
Browse files Browse the repository at this point in the history
* Update test worker image to kubeflow-ci

Related kubeflow/testing#90

* Replace all occurences of mlkube-testing
  • Loading branch information
Ankush Agarwal authored and Penghui Yan committed Jun 18, 2018
1 parent 033c46e commit c863e66
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 14 deletions.
16 changes: 8 additions & 8 deletions examples/gke/test_notebook.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,17 +127,17 @@ def run_test(project, zone, cluster, new_values): # pylint: disable=too-many-lo
def testCpu(self):
"""Test using CPU only."""
now = datetime.datetime.now()
project = "mlkube-testing"
project = "kubeflow-ci"
cluster = ("gke-nb-test-" + now.strftime("v%Y%m%d") + "-"
+ uuid.uuid4().hex[0:4])
zone = "us-east1-d"
new_values = {
"project": project,
"cluster_name": cluster,
"zone": zone,
"registry": "gcr.io/mlkube-testing",
"data_dir": "gs://mlkube-testing_temp/cifar10/data",
"job_dirs": "gs://mlkube-testing_temp/cifar10/jobs",
"registry": "gcr.io/kubeflow-ci",
"data_dir": "gs://kubeflow-ci_temp/cifar10/data",
"job_dirs": "gs://kubeflow-ci_temp/cifar10/jobs",
"num_steps": 10,
"use_gpu": False,
}
Expand All @@ -146,17 +146,17 @@ def testCpu(self):
def testGpu(self):
"""Test using CPU only."""
now = datetime.datetime.now()
project = "mlkube-testing"
project = "kubeflow-ci"
cluster = ("gke-nb-test-" + now.strftime("v%Y%m%d") + "-"
+ uuid.uuid4().hex[0:4])
zone = "us-east1-c"
new_values = {
"project": project,
"cluster_name": cluster,
"zone": zone,
"registry": "gcr.io/mlkube-testing",
"data_dir": "gs://mlkube-testing_temp/cifar10/data",
"job_dirs": "gs://mlkube-testing_temp/cifar10/jobs",
"registry": "gcr.io/kubeflow-ci",
"data_dir": "gs://kubeflow-ci_temp/cifar10/data",
"job_dirs": "gs://kubeflow-ci_temp/cifar10/jobs",
"num_steps": 10,
"use_gpu": True,
"accelerator": "nvidia-tesla-k80",
Expand Down
2 changes: 1 addition & 1 deletion py/prow.py
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ def get_commit_from_env():
def create_latest(gcs_client, job_name, sha):
"""Create a file in GCS with information about the latest passing postsubmit.
"""
bucket_name = "mlkube-testing-results"
bucket_name = "kubeflow-ci-results"
path = os.path.join(job_name, "latest_green.json")

bucket = gcs_client.get_bucket(bucket_name)
Expand Down
4 changes: 2 additions & 2 deletions py/release.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
REPO_ORG = os.getenv("REPO_OWNER", "kubeflow")
REPO_NAME = os.getenv("REPO_NAME", "tf-operator")

RESULTS_BUCKET = "mlkube-testing-results"
RESULTS_BUCKET = "kubeflow-ci-results"
JOB_NAME = "tf-k8s-postsubmit"

GCB_PROJECT = "tf-on-k8s-releasing"
Expand Down Expand Up @@ -542,7 +542,7 @@ def add_common_args(parser):

parser.add_argument(
"--registry",
default="gcr.io/mlkube-testing",
default="gcr.io/kubeflow-ci",
type=str,
help="The docker registry to use.")

Expand Down
5 changes: 2 additions & 3 deletions test/workflows/components/workflows.libsonnet
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
)
else [],


// default parameters.
defaultParams:: {
project:: "kubeflow-ci",
Expand Down Expand Up @@ -54,15 +53,15 @@
local srcRootDir = testDir + "/src";
// The directory containing the kubeflow/tf-operator repo
local srcDir = srcRootDir + "/kubeflow/tf-operator";
local image = "gcr.io/mlkube-testing/test-worker";
local image = "gcr.io/kubeflow-ci/test-worker";
// The name of the NFS volume claim to use for test files.
// local nfsVolumeClaim = "kubeflow-testing";
local nfsVolumeClaim = "nfs-external";
// The name to use for the volume to use to contain test data.
local dataVolume = "kubeflow-test-volume";
local versionTag = if params.versionTag != null then
params.versionTag
else name;
else name;
local tfJobImage = params.registry + "/tf_operator:" + versionTag;

// The namespace on the cluster we spin up to deploy into.
Expand Down

0 comments on commit c863e66

Please sign in to comment.