From bcf2f086f8c6c41e9467f8260581b06ec3bb3a77 Mon Sep 17 00:00:00 2001 From: Jeremy Lewi Date: Wed, 25 Oct 2017 16:22:42 -0700 Subject: [PATCH] Update test infrastructure to use repo tensorflow/k8s (#87) * Update test infrastructure to use repo tensorflow/k8s * Update README to refer to new repo. * Rename go packages from jlewi/mlkube.io > tensorflow/k8s. * Update repo location in the helm e2e test. * Update links. * Update the readme.md --- .gitignore | 4 +++- README.md | 18 +++++++++--------- cmd/tf_operator/main.go | 16 ++++++++-------- developer_guide.md | 4 ++-- images/tf_operator/build_and_push.py | 4 ++-- pkg/controller/controller.go | 8 ++++---- pkg/controller/util.go | 2 +- pkg/garbagecollection/gc.go | 4 ++-- pkg/spec/tf_job.go | 2 +- pkg/spec/tf_job_test.go | 2 +- pkg/trainer/replicas.go | 6 +++--- pkg/trainer/replicas_test.go | 4 ++-- pkg/trainer/tensorboard.go | 2 +- pkg/trainer/tensorboard_test.go | 4 ++-- pkg/trainer/training.go | 10 +++++----- pkg/trainer/training_test.go | 4 ++-- pkg/util/k8sutil/election/election.go | 2 +- pkg/util/k8sutil/fake/fake.go | 2 +- pkg/util/k8sutil/k8sutil.go | 2 +- pkg/util/k8sutil/tpr_util.go | 4 ++-- rename.sh | 2 +- test-infra/README.md | 14 +++++++------- test-infra/helm-test/main.go | 2 +- test-infra/image/bootstrap.py | 6 +++--- test-infra/runner.py | 4 ++-- test/e2e/main.go | 8 ++++---- 26 files changed, 71 insertions(+), 69 deletions(-) diff --git a/.gitignore b/.gitignore index ee25f72242..79ae76af20 100644 --- a/.gitignore +++ b/.gitignore @@ -27,4 +27,6 @@ vendor/ **/bazel-* # Examples egg examples/tf_sample/tf_sample.egg-info/ -examples/.ipynb_checkpoints/ \ No newline at end of file +examples/.ipynb_checkpoints/ + +**/.ipynb_checkpoints diff --git a/README.md b/README.md index 51f0347d35..162c581f52 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,10 @@ # K8s Custom Resource and Operator For TensorFlow jobs -[![Build Status](https://travis-ci.org/jlewi/mlkube.io.svg?branch=master)](https://travis-ci.org/jlewi/mlkube.io) +[![Build Status](https://travis-ci.org/tensorflow/k8s.svg?branch=master)](https://travis-ci.org/tensorflow/k8s) [Prow Test Dashboard](https://k8s-testgrid.appspot.com/sig-big-data) -[Prow Jobs](https://prow.k8s.io/?repo=jlewi%2Fmlkube.io) +[Prow Jobs](https://prow.k8s.io/?repo=tensorflow%2Fk8s) ## Overview @@ -85,15 +85,15 @@ Custom Resources require Kubernetes >= 1.7 ### Configuring the CRD The CRD can be configured via a [ConfigMap](https://kubernetes.io/docs/api-reference/v1.8/#configmap-v1-core) -that provides a [ControllerConfig](https://github.com/jlewi/mlkube.io/blob/master/pkg/spec/controller.go) serialized +that provides a [ControllerConfig](https://github.com/tensorflow/k8s/blob/master/pkg/spec/controller.go) serialized as YAML. The config controls how the CRD manages TensorFlow jobs. -Currently, the most important use for [ControllerConfig](https://github.com/jlewi/mlkube.io/blob/master/pkg/spec/controller.go) +Currently, the most important use for [ControllerConfig](https://github.com/tensorflow/k8s/blob/master/pkg/spec/controller.go) is specifying environment variables and volumes that must be mounted from the host into containers to configure GPUS. The TfJob controller can be configured with a list of volumes that should be mounted from the host into the container -to make GPUs work. Here's an example [ControllerConfig](https://github.com/jlewi/mlkube.io/blob/master/pkg/spec/controller.go): +to make GPUs work. Here's an example [ControllerConfig](https://github.com/tensorflow/k8s/blob/master/pkg/spec/controller.go): ``` accelerators: @@ -119,7 +119,7 @@ The helm package for the controller includes a config map suitable for GKE. This ConfigMap may need to be modified for your cluster if you aren't using GKE. -There's an open [issue](https://github.com/jlewi/mlkube.io/issues/71) to +There's an open [issue](https://github.com/tensorflow/k8s/issues/71) to better support non GKE clusters @@ -128,7 +128,7 @@ better support non GKE clusters You create a job by defining a TfJob and then creating it with. ``` -kubectl create -f https://raw.githubusercontent.com/jlewi/mlkube.io/master/examples/tf_job.yaml +kubectl create -f https://raw.githubusercontent.com/tensorflow/k8s/master/examples/tf_job.yaml ``` In this case the job spec looks like the following @@ -234,9 +234,9 @@ for using GPUs. ### Requesting a TensorBoard instance You can also ask the `TfJob` operator to create a TensorBoard instance -by including a [TensorBoardSpec](https://github.com/jlewi/mlkube.io/blob/master/pkg/spec/tf_job.go#L103) +by including a [TensorBoardSpec](https://github.com/tensorflow/k8s/blob/master/pkg/spec/tf_job.go#L103) in your job. The table below describes the important fields in -[TensorBoardSpec](https://github.com/jlewi/mlkube.io/blob/master/pkg/spec/tf_job.go#L103). +[TensorBoardSpec](https://github.com/tensorflow/k8s/blob/master/pkg/spec/tf_job.go#L103). | Name | Description | Required | Default | |---|---|---|---| diff --git a/cmd/tf_operator/main.go b/cmd/tf_operator/main.go index 824cebe0c6..5fe2db6400 100644 --- a/cmd/tf_operator/main.go +++ b/cmd/tf_operator/main.go @@ -10,19 +10,19 @@ import ( "github.com/ghodss/yaml" - "github.com/jlewi/mlkube.io/pkg/controller" - "github.com/jlewi/mlkube.io/pkg/garbagecollection" - "github.com/jlewi/mlkube.io/pkg/util" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil/election" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil/election/resourcelock" - "github.com/jlewi/mlkube.io/version" + "github.com/tensorflow/k8s/pkg/controller" + "github.com/tensorflow/k8s/pkg/garbagecollection" + "github.com/tensorflow/k8s/pkg/util" + "github.com/tensorflow/k8s/pkg/util/k8sutil" + "github.com/tensorflow/k8s/pkg/util/k8sutil/election" + "github.com/tensorflow/k8s/pkg/util/k8sutil/election/resourcelock" + "github.com/tensorflow/k8s/version" log "github.com/golang/glog" "io/ioutil" - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" "k8s.io/client-go/kubernetes" "k8s.io/client-go/pkg/api/v1" "k8s.io/client-go/tools/record" diff --git a/developer_guide.md b/developer_guide.md index 8e057947c2..902a5a576f 100644 --- a/developer_guide.md +++ b/developer_guide.md @@ -8,7 +8,7 @@ Create a symbolic link inside your GOPATH to the location you checked out the co ln -sf ${GIT_TRAINING} ${GOPATH}/src/mlkube.io ``` - * GIT_TRAINING should be the location where you checked out https://github.com/jlewi/mlkube.io + * GIT_TRAINING should be the location where you checked out https://github.com/tensorflow/k8s Resolve dependencies (if you don't have glide install, check how to do it [here](https://github.com/Masterminds/glide/blob/master/README.md#install)) @@ -24,7 +24,7 @@ rm -rf vendor/k8s.io/apiextensions-apiserver/vendor Build it ``` -go install github.com/jlewi/mlkube.io/cmd/tf_operator +go install github.com/tensorflow/k8s/cmd/tf_operator ``` ## Runing the Operator Locally diff --git a/images/tf_operator/build_and_push.py b/images/tf_operator/build_and_push.py index cc5efd9f03..d459bf9199 100755 --- a/images/tf_operator/build_and_push.py +++ b/images/tf_operator/build_and_push.py @@ -95,8 +95,8 @@ def run_and_output(command, cwd=None): go_path = os.environ["GOPATH"] targets = [ - "github.com/jlewi/mlkube.io/cmd/tf_operator", - "github.com/jlewi/mlkube.io/test/e2e", + "github.com/tensorflow/k8s/cmd/tf_operator", + "github.com/tensorflow/k8s/test/e2e", ] for t in targets: subprocess.check_call(["go", "install", t]) diff --git a/pkg/controller/controller.go b/pkg/controller/controller.go index 65ca498f36..d258148472 100644 --- a/pkg/controller/controller.go +++ b/pkg/controller/controller.go @@ -11,13 +11,13 @@ import ( "sync" "time" - "github.com/jlewi/mlkube.io/pkg/spec" - "github.com/jlewi/mlkube.io/pkg/trainer" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil" + "github.com/tensorflow/k8s/pkg/spec" + "github.com/tensorflow/k8s/pkg/trainer" + "github.com/tensorflow/k8s/pkg/util/k8sutil" "k8s.io/client-go/kubernetes" log "github.com/golang/glog" - "github.com/jlewi/mlkube.io/pkg/util" + "github.com/tensorflow/k8s/pkg/util" v1beta1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1beta1" apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset" apierrors "k8s.io/apimachinery/pkg/api/errors" diff --git a/pkg/controller/util.go b/pkg/controller/util.go index 24d45475d0..6d33da1fc5 100644 --- a/pkg/controller/util.go +++ b/pkg/controller/util.go @@ -6,7 +6,7 @@ import ( "io" "time" - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" kwatch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/garbagecollection/gc.go b/pkg/garbagecollection/gc.go index dcc0f5ee58..0eab5142aa 100644 --- a/pkg/garbagecollection/gc.go +++ b/pkg/garbagecollection/gc.go @@ -13,8 +13,8 @@ package garbagecollection import ( - "github.com/jlewi/mlkube.io/pkg/spec" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil" + "github.com/tensorflow/k8s/pkg/spec" + "github.com/tensorflow/k8s/pkg/util/k8sutil" log "github.com/golang/glog" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" diff --git a/pkg/spec/tf_job.go b/pkg/spec/tf_job.go index 63ce7ffd7e..2cd1247bbb 100644 --- a/pkg/spec/tf_job.go +++ b/pkg/spec/tf_job.go @@ -7,7 +7,7 @@ import ( "time" "github.com/golang/protobuf/proto" - "github.com/jlewi/mlkube.io/pkg/util" + "github.com/tensorflow/k8s/pkg/util" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/client-go/pkg/api/v1" ) diff --git a/pkg/spec/tf_job_test.go b/pkg/spec/tf_job_test.go index cb563362b4..485d55bfc9 100644 --- a/pkg/spec/tf_job_test.go +++ b/pkg/spec/tf_job_test.go @@ -5,7 +5,7 @@ import ( "testing" "github.com/gogo/protobuf/proto" - "github.com/jlewi/mlkube.io/pkg/util" + "github.com/tensorflow/k8s/pkg/util" "k8s.io/apimachinery/pkg/api/resource" "k8s.io/client-go/pkg/api/v1" ) diff --git a/pkg/trainer/replicas.go b/pkg/trainer/replicas.go index 20b93e4db7..00359efdde 100644 --- a/pkg/trainer/replicas.go +++ b/pkg/trainer/replicas.go @@ -8,14 +8,14 @@ import ( "sort" "strings" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil" + "github.com/tensorflow/k8s/pkg/util/k8sutil" - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" log "github.com/golang/glog" "github.com/golang/protobuf/proto" // TOOO(jlewi): Rename to apiErrors - "github.com/jlewi/mlkube.io/pkg/util" + "github.com/tensorflow/k8s/pkg/util" k8s_errors "k8s.io/apimachinery/pkg/api/errors" meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1" k8sErrors "k8s.io/apimachinery/pkg/util/errors" diff --git a/pkg/trainer/replicas_test.go b/pkg/trainer/replicas_test.go index 9a31646f80..26045d7b6f 100644 --- a/pkg/trainer/replicas_test.go +++ b/pkg/trainer/replicas_test.go @@ -10,8 +10,8 @@ import ( "sync" "time" - "github.com/jlewi/mlkube.io/pkg/spec" - tfJobFake "github.com/jlewi/mlkube.io/pkg/util/k8sutil/fake" + "github.com/tensorflow/k8s/pkg/spec" + tfJobFake "github.com/tensorflow/k8s/pkg/util/k8sutil/fake" meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/client-go/kubernetes/fake" "k8s.io/client-go/pkg/api/v1" diff --git a/pkg/trainer/tensorboard.go b/pkg/trainer/tensorboard.go index 8b1998b9d2..958102252f 100644 --- a/pkg/trainer/tensorboard.go +++ b/pkg/trainer/tensorboard.go @@ -7,7 +7,7 @@ import ( log "github.com/golang/glog" "github.com/golang/protobuf/proto" - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" k8s_errors "k8s.io/apimachinery/pkg/api/errors" meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/util/intstr" diff --git a/pkg/trainer/tensorboard_test.go b/pkg/trainer/tensorboard_test.go index 5be087d242..76cb60938b 100644 --- a/pkg/trainer/tensorboard_test.go +++ b/pkg/trainer/tensorboard_test.go @@ -8,8 +8,8 @@ import ( "reflect" "sync" - "github.com/jlewi/mlkube.io/pkg/spec" - tfJobFake "github.com/jlewi/mlkube.io/pkg/util/k8sutil/fake" + "github.com/tensorflow/k8s/pkg/spec" + tfJobFake "github.com/tensorflow/k8s/pkg/util/k8sutil/fake" meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/client-go/kubernetes/fake" "k8s.io/client-go/pkg/api/v1" diff --git a/pkg/trainer/training.go b/pkg/trainer/training.go index d40009480e..d5048aa7a9 100644 --- a/pkg/trainer/training.go +++ b/pkg/trainer/training.go @@ -7,17 +7,17 @@ import ( "reflect" log "github.com/golang/glog" - "github.com/jlewi/mlkube.io/pkg/spec" - "github.com/jlewi/mlkube.io/pkg/util" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil" - "github.com/jlewi/mlkube.io/pkg/util/retryutil" + "github.com/tensorflow/k8s/pkg/spec" + "github.com/tensorflow/k8s/pkg/util" + "github.com/tensorflow/k8s/pkg/util/k8sutil" + "github.com/tensorflow/k8s/pkg/util/retryutil" "math" "strings" "sync" "time" - "github.com/jlewi/mlkube.io/pkg/garbagecollection" + "github.com/tensorflow/k8s/pkg/garbagecollection" apierrors "k8s.io/apimachinery/pkg/api/errors" "k8s.io/client-go/kubernetes" diff --git a/pkg/trainer/training_test.go b/pkg/trainer/training_test.go index f7166f2ec1..a5d0eb0eb3 100644 --- a/pkg/trainer/training_test.go +++ b/pkg/trainer/training_test.go @@ -8,8 +8,8 @@ import ( "k8s.io/apimachinery/pkg/api/resource" "k8s.io/client-go/kubernetes/fake" "k8s.io/client-go/pkg/api/v1" - "github.com/jlewi/mlkube.io/pkg/spec" - tfJobFake "github.com/jlewi/mlkube.io/pkg/util/k8sutil/fake" + "github.com/tensorflow/k8s/pkg/spec" + tfJobFake "github.com/tensorflow/k8s/pkg/util/k8sutil/fake" "sync" ) diff --git a/pkg/util/k8sutil/election/election.go b/pkg/util/k8sutil/election/election.go index 4a2141a0d3..8d5249d2f4 100644 --- a/pkg/util/k8sutil/election/election.go +++ b/pkg/util/k8sutil/election/election.go @@ -59,7 +59,7 @@ import ( "k8s.io/apimachinery/pkg/util/wait" "github.com/golang/glog" - rl "github.com/jlewi/mlkube.io/pkg/util/k8sutil/election/resourcelock" + rl "github.com/tensorflow/k8s/pkg/util/k8sutil/election/resourcelock" ) const ( diff --git a/pkg/util/k8sutil/fake/fake.go b/pkg/util/k8sutil/fake/fake.go index 0d97ee915b..13511839a0 100644 --- a/pkg/util/k8sutil/fake/fake.go +++ b/pkg/util/k8sutil/fake/fake.go @@ -2,7 +2,7 @@ package fake import ( - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" "net/http" "time" ) diff --git a/pkg/util/k8sutil/k8sutil.go b/pkg/util/k8sutil/k8sutil.go index bd863c433b..36aacbefb3 100644 --- a/pkg/util/k8sutil/k8sutil.go +++ b/pkg/util/k8sutil/k8sutil.go @@ -4,7 +4,7 @@ import ( "net" "os" - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset" apierrors "k8s.io/apimachinery/pkg/api/errors" diff --git a/pkg/util/k8sutil/tpr_util.go b/pkg/util/k8sutil/tpr_util.go index ec69a8a985..01cb27d0f8 100644 --- a/pkg/util/k8sutil/tpr_util.go +++ b/pkg/util/k8sutil/tpr_util.go @@ -19,12 +19,12 @@ import ( "fmt" "net/http" - "github.com/jlewi/mlkube.io/pkg/spec" + "github.com/tensorflow/k8s/pkg/spec" "k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/runtime/serializer" "k8s.io/client-go/pkg/api" "k8s.io/client-go/rest" - "github.com/jlewi/mlkube.io/pkg/util" + "github.com/tensorflow/k8s/pkg/util" log "github.com/golang/glog" ) diff --git a/rename.sh b/rename.sh index 3140208889..3d975855e3 100755 --- a/rename.sh +++ b/rename.sh @@ -3,5 +3,5 @@ # Rewrite some imports files=`find ./ -name *.go` for f in $files; do - sed -i "s/mlkube.io\//github.com\/jlewi\/mlkube.io\//" ${f} + sed -i "s/github.com\/jlewi\/mlkube.io\//github.com\/tensorflow\/k8s\//" ${f} done diff --git a/test-infra/README.md b/test-infra/README.md index 537fead3e7..68b56a9c9d 100644 --- a/test-infra/README.md +++ b/test-infra/README.md @@ -1,4 +1,4 @@ -# mlkube Test Infrastructure +# Test Infrastructure We use [Prow](https://github.com/kubernetes/test-infra/tree/master/prow), K8s' continuous integration tool. @@ -15,9 +15,9 @@ We use Prow to run: Quick Links * [config.yaml](https://github.com/kubernetes/test-infra/blob/master/prow/config.yaml) defines the ProwJobs. - * Search for mlkube to find mlkube related jobs - * [mlkube Test Results Dashboard](https://k8s-testgrid.appspot.com/sig-big-data) - * [mlkube Prow Jobs dashboard](https://prow.k8s.io/?repo=jlewi%2Fmlkube.io) + * Search for tf-k8s to find tf-k8s related jobs + * [tf-k8s Test Results Dashboard](https://k8s-testgrid.appspot.com/sig-big-data) + * [tf-k8s Prow Jobs dashboard](https://prow.k8s.io/?repo=tensorflow%2Fk8s) ## Anatomy of our Prow Jobs @@ -63,7 +63,7 @@ You can also run the tests inside the Docker image, * This can be useful for debugging or testing changes ``` - docker run -ti -v ${REPO_PATH}:/go/src/github.com/jlewi/mlkube.io \ + docker run -ti -v ${REPO_PATH}:/go/src/github.com/tensorflow/k8s \ -v /var/run/docker.sock:/var/run/docker.sock \ --entrypoint=/bin/bash gcr.io/mlkube-testing/builder:latest gcloud auth login @@ -130,7 +130,7 @@ the results. Our jobs should be added to [K8s config](https://github.com/kubernetes/test-infra/blob/master/prow/config.yaml) -## Notes adding mlkube.io to K8s Prow Instance +## Notes adding tensorflow/k8s to K8s Prow Instance Below is some notes on what it took to integrate with K8s Prow instance. @@ -141,7 +141,7 @@ Below is some notes on what it took to integrate with K8s Prow instance. * Add test dashboards to [testgrid/config/config.yaml](https://github.com/kubernetes/test-infra/pull/4951/files#diff-49f154cd90facc43fda49a99885e6d17) * Modify [testgrid/jenkins_verify/jenkins_validat.go](https://github.com/kubernetes/test-infra/pull/4951/files#diff-7fb4731a02dd681bbd0daada8dd2f908) to allow presubmits for the new repo. -1. For mlkube.io configure webhooks by following these [instructions](https://github.com/kubernetes/test-infra/blob/master/prow/getting_started.md#add-the-webhook-to-github) +1. For tensorflow/k8s configure webhooks by following these [instructions](https://github.com/kubernetes/test-infra/blob/master/prow/getting_started.md#add-the-webhook-to-github) * Use https://prow.k8s.io/hook as the target * Get HMAC token from k8s test team 1. Add the k8s bot account, k8s-ci-robot, as an admin on the repository diff --git a/test-infra/helm-test/main.go b/test-infra/helm-test/main.go index ae6faf057b..de331057df 100644 --- a/test-infra/helm-test/main.go +++ b/test-infra/helm-test/main.go @@ -107,7 +107,7 @@ var ( TEST_FAILURE_CODE = 2 // File path constants - chartsBasePath = path.Join(os.Getenv("GOPATH"), "src", "/github.com/jlewi/mlkube.io") + chartsBasePath = path.Join(os.Getenv("GOPATH"), "src", "/github.com/tensorflow/k8s") image = flag.String("image", "", "The Docker image for Tfjob to use.") outputPath = flag.String("output_dir", "", "The directory where test output should be written.") diff --git a/test-infra/image/bootstrap.py b/test-infra/image/bootstrap.py index 200c0e75ba..4f7a2ff9a9 100644 --- a/test-infra/image/bootstrap.py +++ b/test-infra/image/bootstrap.py @@ -28,13 +28,13 @@ # Default name for the repo organization and name. # This should match the values used in Go imports. -GO_REPO_OWNER = "jlewi" -GO_REPO_NAME = "mlkube.io" +GO_REPO_OWNER = "tensorflow" +GO_REPO_NAME = "k8s" def run(command, cwd=None): logging.info("Running: %s", " ".join(command)) - subprocess.check_call(command, cwd=cwd).decode("utf-8") + subprocess.check_call(command, cwd=cwd) def run_and_output(command, cwd=None): logging.info("Running: %s", " ".join(command)) diff --git a/test-infra/runner.py b/test-infra/runner.py index 97e222ca24..bdf552ee80 100644 --- a/test-infra/runner.py +++ b/test-infra/runner.py @@ -54,8 +54,8 @@ # Default repository organization and name. # This should match the values used in Go imports. -GO_REPO_OWNER = "jlewi" -GO_REPO_NAME = "mlkube.io" +GO_REPO_OWNER = "tensorflow" +GO_REPO_NAME = "k8s" GCS_REGEX = re.compile("gs://([^/]*)/(.*)") diff --git a/test/e2e/main.go b/test/e2e/main.go index ac0685b535..2eae4aaf15 100644 --- a/test/e2e/main.go +++ b/test/e2e/main.go @@ -7,7 +7,7 @@ // TODO(jlewi): Do we need to make the test output conform to the TAP(https://testanything.org/) // protocol so we can fit into the K8s dashboard // -// TODO(https://github.com/jlewi/mlkube.io/issues/21) The E2E test should actually run distributed TensorFlow. +// TODO(https://github.com/tensorflow/k8s/issues/21) The E2E test should actually run distributed TensorFlow. package main import ( @@ -19,9 +19,9 @@ import ( "github.com/gogo/protobuf/proto" log "github.com/golang/glog" - "github.com/jlewi/mlkube.io/pkg/spec" - "github.com/jlewi/mlkube.io/pkg/util" - "github.com/jlewi/mlkube.io/pkg/util/k8sutil" + "github.com/tensorflow/k8s/pkg/spec" + "github.com/tensorflow/k8s/pkg/util" + "github.com/tensorflow/k8s/pkg/util/k8sutil" k8s_errors "k8s.io/apimachinery/pkg/api/errors" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/client-go/pkg/api/v1"