diff --git a/README.md b/README.md index 57e8625a32b..b9000e12653 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ # Katib -[![Go Report Card](https://goreportcard.com/badge/github.com/kubeflow/hp-tuning)](https://goreportcard.com/report/github.com/kubeflow/hp-tuning) +[![Go Report Card](https://goreportcard.com/badge/github.com/kubeflow/katib)](https://goreportcard.com/report/github.com/kubeflow/katib) Hyperparameter Tuning on Kubernetes. This project is inspired by [Google vizier](https://static.googleusercontent.com/media/research.google.com/ja//pubs/archive/bcb15507f4b52991a0783013df4222240e942381.pdf). Katib is a scalable and flexible hyperparameter tuning framework and is tightly integrated with kubernetes. Also it does not depend on a specific Deep Learning framework e.g. TensorFlow, MXNet, and PyTorch). diff --git a/cli/Dockerfile b/cli/Dockerfile index 9ea39d52397..4d8c88473d3 100644 --- a/cli/Dockerfile +++ b/cli/Dockerfile @@ -1,9 +1,9 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/cli +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/cli RUN go build -o katib-cli FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/cli/katib-cli /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/cli/katib-cli /app/ diff --git a/cli/main.go b/cli/main.go index 061c0db735d..66640b29d87 100644 --- a/cli/main.go +++ b/cli/main.go @@ -12,7 +12,7 @@ import ( "google.golang.org/grpc" - pb "github.com/kubeflow/hp-tuning/api" + pb "github.com/kubeflow/katib/api" ) var server = flag.String("s", "127.0.0.1:6789", "server address") diff --git a/db/interface.go b/db/interface.go index c1c0738491e..f83eba8b018 100644 --- a/db/interface.go +++ b/db/interface.go @@ -12,7 +12,7 @@ import ( "strings" "time" - api "github.com/kubeflow/hp-tuning/api" + api "github.com/kubeflow/katib/api" _ "github.com/go-sql-driver/mysql" ) diff --git a/db/interface_test.go b/db/interface_test.go index 91de1eaeec8..0bb32bcaa0a 100644 --- a/db/interface_test.go +++ b/db/interface_test.go @@ -14,7 +14,7 @@ import ( "github.com/golang/protobuf/jsonpb" "gopkg.in/DATA-DOG/go-sqlmock.v1" - api "github.com/kubeflow/hp-tuning/api" + api "github.com/kubeflow/katib/api" ) var db_interface VizierDBInterface diff --git a/db/test/test.go b/db/test/test.go index 638ae3e4713..4aa5174f4d1 100644 --- a/db/test/test.go +++ b/db/test/test.go @@ -2,7 +2,7 @@ package main import ( "fmt" - "github.com/kubeflow/hp-tuning/db" + "github.com/kubeflow/katib/db" "os" ) diff --git a/dlk/Dockerfile b/dlk/Dockerfile index 6fa23100745..6e4eff78afe 100644 --- a/dlk/Dockerfile +++ b/dlk/Dockerfile @@ -1,10 +1,10 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/dlk/dlkmanager -RUN go build -o dlkmanager +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/dlk/dlkmanager +RUN go build -o dlkmanager FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/dlk/dlkmanager/dlkmanager /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/dlk/dlkmanager/dlkmanager /app/ ENTRYPOINT ["./dlkmanager"] diff --git a/dlk/dlkctl/del-lt.go b/dlk/dlkctl/del-lt.go index c24f848ab19..39420eb4f2a 100644 --- a/dlk/dlkctl/del-lt.go +++ b/dlk/dlkctl/del-lt.go @@ -19,7 +19,7 @@ import ( "net/http" "os" - "github.com/kubeflow/hp-tuning/dlk/dlkctl/utils" + "github.com/kubeflow/katib/dlk/dlkctl/utils" "github.com/spf13/cobra" ) diff --git a/dlk/dlkctl/get-lt.go b/dlk/dlkctl/get-lt.go index 8e251e73d75..20bc6a26fd5 100644 --- a/dlk/dlkctl/get-lt.go +++ b/dlk/dlkctl/get-lt.go @@ -22,9 +22,9 @@ import ( "os" "sort" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" - "github.com/kubeflow/hp-tuning/dlk/dlkctl/utils" + "github.com/kubeflow/katib/dlk/dlkctl/utils" "github.com/spf13/cobra" ) diff --git a/dlk/dlkctl/get-lts.go b/dlk/dlkctl/get-lts.go index a591c63b3f8..4e66e932d18 100644 --- a/dlk/dlkctl/get-lts.go +++ b/dlk/dlkctl/get-lts.go @@ -21,9 +21,9 @@ import ( "net/http" "os" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" - "github.com/kubeflow/hp-tuning/dlk/dlkctl/utils" + "github.com/kubeflow/katib/dlk/dlkctl/utils" "github.com/spf13/cobra" ) diff --git a/dlk/dlkctl/logs.go b/dlk/dlkctl/logs.go index dde83b8be2a..03b63effe06 100644 --- a/dlk/dlkctl/logs.go +++ b/dlk/dlkctl/logs.go @@ -23,8 +23,8 @@ import ( "os" "time" - "github.com/kubeflow/hp-tuning/dlk/dlkctl/utils" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkctl/utils" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" "github.com/spf13/cobra" ) diff --git a/dlk/dlkctl/run.go b/dlk/dlkctl/run.go index 46aa3ffa2d5..0ff9db017a0 100644 --- a/dlk/dlkctl/run.go +++ b/dlk/dlkctl/run.go @@ -19,8 +19,8 @@ import ( "strings" "time" - "github.com/kubeflow/hp-tuning/dlk/dlkctl/utils" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/api" + "github.com/kubeflow/katib/dlk/dlkctl/utils" + "github.com/kubeflow/katib/dlk/dlkmanager/api" "github.com/spf13/cobra" ) diff --git a/dlk/dlkmanager/delete.go b/dlk/dlkmanager/delete.go index b04ed02fb06..ba8585dbf35 100644 --- a/dlk/dlkmanager/delete.go +++ b/dlk/dlkmanager/delete.go @@ -4,7 +4,7 @@ import ( "fmt" "net/http" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" "github.com/labstack/echo" lgr "github.com/sirupsen/logrus" ) diff --git a/dlk/dlkmanager/dlkmanager.go b/dlk/dlkmanager/dlkmanager.go index b6d5e57abed..e200e427df4 100644 --- a/dlk/dlkmanager/dlkmanager.go +++ b/dlk/dlkmanager/dlkmanager.go @@ -4,7 +4,7 @@ import ( "fmt" "os" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/configs" + "github.com/kubeflow/katib/dlk/dlkmanager/configs" "github.com/labstack/echo" // Web server framework for REST API lgr "github.com/sirupsen/logrus" // logging framework "github.com/spf13/cobra" diff --git a/dlk/dlkmanager/get.go b/dlk/dlkmanager/get.go index 1589ffb6a4e..d25e31287d4 100644 --- a/dlk/dlkmanager/get.go +++ b/dlk/dlkmanager/get.go @@ -4,7 +4,7 @@ import ( "net/http" "time" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" "github.com/labstack/echo" lgr "github.com/sirupsen/logrus" ) diff --git a/dlk/dlkmanager/learningTask.go b/dlk/dlkmanager/learningTask.go index dec6b942361..6be465add93 100644 --- a/dlk/dlkmanager/learningTask.go +++ b/dlk/dlkmanager/learningTask.go @@ -7,8 +7,8 @@ import ( "sync" "time" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/api" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkmanager/api" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" lgr "github.com/sirupsen/logrus" batchv1 "k8s.io/api/batch/v1" diff --git a/dlk/dlkmanager/post.go b/dlk/dlkmanager/post.go index 8661a8fae4c..7ca55a1175f 100644 --- a/dlk/dlkmanager/post.go +++ b/dlk/dlkmanager/post.go @@ -6,10 +6,10 @@ import ( "path/filepath" "time" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/configs" + "github.com/kubeflow/katib/dlk/dlkmanager/configs" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/api" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" + "github.com/kubeflow/katib/dlk/dlkmanager/api" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" corev1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/client-go/kubernetes" diff --git a/docs/getting-start.md b/docs/getting-start.md index 350d9a4d478..8454f7c2611 100644 --- a/docs/getting-start.md +++ b/docs/getting-start.md @@ -11,7 +11,7 @@ First, Copy CLI tool. ```bash -$ curl -Lo katib-cli https://github.com/kubeflow/hp-tuning/releases/download/v0.1.0-alpha/katib-cli-linux-amd64 && chmod +x katib-cli && sudo mv katib-cli /usr/local/bin/ +$ curl -Lo katib-cli https://github.com/kubeflow/katib/releases/download/v0.1.0-alpha/katib-cli-linux-amd64 && chmod +x katib-cli && sudo mv katib-cli /usr/local/bin/ ``` The cli tool will be put `/usr/local/bin/` directory. diff --git a/earlystopping/medianstopping.go b/earlystopping/medianstopping.go index 57e2d86ae66..6c07ab32013 100644 --- a/earlystopping/medianstopping.go +++ b/earlystopping/medianstopping.go @@ -3,8 +3,8 @@ package earlystopping import ( "context" "errors" - "github.com/kubeflow/hp-tuning/api" - vdb "github.com/kubeflow/hp-tuning/db" + "github.com/kubeflow/katib/api" + vdb "github.com/kubeflow/katib/db" "log" "sort" "strconv" diff --git a/earlystopping/medianstopping/Dockerfile b/earlystopping/medianstopping/Dockerfile index 3daede6fd9a..372254494ed 100644 --- a/earlystopping/medianstopping/Dockerfile +++ b/earlystopping/medianstopping/Dockerfile @@ -1,10 +1,10 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/earlystopping/medianstopping +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/earlystopping/medianstopping RUN go build -o medianstopping FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/earlystopping/medianstopping /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/earlystopping/medianstopping /app/ ENTRYPOINT ["./medianstopping"] diff --git a/earlystopping/medianstopping/main.go b/earlystopping/medianstopping/main.go index 6244acb3dee..a3da6c333a9 100644 --- a/earlystopping/medianstopping/main.go +++ b/earlystopping/medianstopping/main.go @@ -7,8 +7,8 @@ import ( "google.golang.org/grpc" "google.golang.org/grpc/reflection" - pb "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/earlystopping" + pb "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/earlystopping" ) func main() { diff --git a/earlystopping/types.go b/earlystopping/types.go index e3d85ec7da7..cd5f6dd560b 100644 --- a/earlystopping/types.go +++ b/earlystopping/types.go @@ -3,7 +3,7 @@ package earlystopping import ( "context" - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" ) const ( diff --git a/manager/Dockerfile b/manager/Dockerfile index aac578594aa..23fe2e0ed15 100644 --- a/manager/Dockerfile +++ b/manager/Dockerfile @@ -1,12 +1,12 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/manager +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/manager RUN go build -o vizier-manager FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/manager/vizier-manager /app/ -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/manager/visualise / +COPY --from=build-env /go/src/github.com/kubeflow/katib/manager/vizier-manager /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/manager/visualise / ENTRYPOINT ["./vizier-manager"] CMD ["-w", "dlk"] diff --git a/manager/main.go b/manager/main.go index 91314b12fbd..b8342f2d08e 100644 --- a/manager/main.go +++ b/manager/main.go @@ -9,14 +9,14 @@ import ( "strconv" "time" - pb "github.com/kubeflow/hp-tuning/api" - kdb "github.com/kubeflow/hp-tuning/db" - "github.com/kubeflow/hp-tuning/manager/modelstore" - tbif "github.com/kubeflow/hp-tuning/manager/visualise/tensorboard" - "github.com/kubeflow/hp-tuning/manager/worker_interface" - dlkwif "github.com/kubeflow/hp-tuning/manager/worker_interface/dlk" - k8swif "github.com/kubeflow/hp-tuning/manager/worker_interface/kubernetes" - nvdwif "github.com/kubeflow/hp-tuning/manager/worker_interface/nvdocker" + pb "github.com/kubeflow/katib/api" + kdb "github.com/kubeflow/katib/db" + "github.com/kubeflow/katib/manager/modelstore" + tbif "github.com/kubeflow/katib/manager/visualise/tensorboard" + "github.com/kubeflow/katib/manager/worker_interface" + dlkwif "github.com/kubeflow/katib/manager/worker_interface/dlk" + k8swif "github.com/kubeflow/katib/manager/worker_interface/kubernetes" + nvdwif "github.com/kubeflow/katib/manager/worker_interface/nvdocker" "google.golang.org/grpc" "google.golang.org/grpc/reflection" diff --git a/manager/main_test.go b/manager/main_test.go index 71bc785f4cb..6f5717f47b6 100644 --- a/manager/main_test.go +++ b/manager/main_test.go @@ -6,10 +6,10 @@ import ( "github.com/golang/mock/gomock" - api "github.com/kubeflow/hp-tuning/api" - mockdb "github.com/kubeflow/hp-tuning/mock/db" - mockmodelstore "github.com/kubeflow/hp-tuning/mock/modelstore" - mockworker "github.com/kubeflow/hp-tuning/mock/worker" + api "github.com/kubeflow/katib/api" + mockdb "github.com/kubeflow/katib/mock/db" + mockmodelstore "github.com/kubeflow/katib/mock/modelstore" + mockworker "github.com/kubeflow/katib/mock/worker" ) func TestCreateStudy(t *testing.T) { diff --git a/manager/modelstore/modeldb.go b/manager/modelstore/modeldb.go index 8b179f1e3d4..fd90688ea42 100644 --- a/manager/modelstore/modeldb.go +++ b/manager/modelstore/modeldb.go @@ -4,8 +4,8 @@ import ( "context" "fmt" "git.apache.org/thrift.git/lib/go/thrift" - "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/manager/modelstore/modeldb" + "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/manager/modelstore/modeldb" "log" "net" "strconv" diff --git a/manager/modelstore/modeldb/model_d_b_service-remote/model_d_b_service-remote.go b/manager/modelstore/modeldb/model_d_b_service-remote/model_d_b_service-remote.go index 3f0a100e994..85a358106eb 100755 --- a/manager/modelstore/modeldb/model_d_b_service-remote/model_d_b_service-remote.go +++ b/manager/modelstore/modeldb/model_d_b_service-remote/model_d_b_service-remote.go @@ -8,7 +8,7 @@ import ( "flag" "fmt" "git.apache.org/thrift.git/lib/go/thrift" - "github.com/kubeflow/hp-tuning/manager/modelstore/modeldb" + "github.com/kubeflow/katib/manager/modelstore/modeldb" "math" "net" "net/url" diff --git a/manager/modelstore/modelstore.go b/manager/modelstore/modelstore.go index c2cfdfc382a..a346ab1d416 100644 --- a/manager/modelstore/modelstore.go +++ b/manager/modelstore/modelstore.go @@ -1,7 +1,7 @@ package modelstore import ( - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" ) type ModelStore interface { diff --git a/manager/visualise/tensorboard/tensorboardIF.go b/manager/visualise/tensorboard/tensorboardIF.go index b74f24135fa..7d3d2c04ba7 100644 --- a/manager/visualise/tensorboard/tensorboardIF.go +++ b/manager/visualise/tensorboard/tensorboardIF.go @@ -2,7 +2,7 @@ package tensorboard import ( "bytes" - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" "io/ioutil" apiv1 "k8s.io/api/core/v1" exbeatav1 "k8s.io/api/extensions/v1beta1" diff --git a/manager/worker_interface/dlk/dlk.go b/manager/worker_interface/dlk/dlk.go index cfec1024bc9..8d6b739e3e7 100644 --- a/manager/worker_interface/dlk/dlk.go +++ b/manager/worker_interface/dlk/dlk.go @@ -4,11 +4,11 @@ import ( "bytes" "encoding/json" "fmt" - "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/db" - dlkapi "github.com/kubeflow/hp-tuning/dlk/dlkmanager/api" - "github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore" - wIF "github.com/kubeflow/hp-tuning/manager/worker_interface" + "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/db" + dlkapi "github.com/kubeflow/katib/dlk/dlkmanager/api" + "github.com/kubeflow/katib/dlk/dlkmanager/datastore" + wIF "github.com/kubeflow/katib/manager/worker_interface" "io/ioutil" "log" "net/http" diff --git a/manager/worker_interface/kubernetes/kubernetes.go b/manager/worker_interface/kubernetes/kubernetes.go index 75ac540d09b..8775bde158b 100644 --- a/manager/worker_interface/kubernetes/kubernetes.go +++ b/manager/worker_interface/kubernetes/kubernetes.go @@ -4,10 +4,10 @@ import ( "bytes" "errors" "fmt" - "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/db" - "github.com/kubeflow/hp-tuning/earlystopping" - wIF "github.com/kubeflow/hp-tuning/manager/worker_interface" + "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/db" + "github.com/kubeflow/katib/earlystopping" + wIF "github.com/kubeflow/katib/manager/worker_interface" "io/ioutil" batchv1 "k8s.io/api/batch/v1" apiv1 "k8s.io/api/core/v1" diff --git a/manager/worker_interface/nvdocker/nvidia-docker.go b/manager/worker_interface/nvdocker/nvidia-docker.go index 1dbb79a4c68..a264e0a52bd 100644 --- a/manager/worker_interface/nvdocker/nvidia-docker.go +++ b/manager/worker_interface/nvdocker/nvidia-docker.go @@ -8,9 +8,9 @@ import ( "github.com/docker/docker/api/types" "github.com/docker/docker/api/types/container" dclient "github.com/docker/docker/client" - "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/db" - wIF "github.com/kubeflow/hp-tuning/manager/worker_interface" + "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/db" + wIF "github.com/kubeflow/katib/manager/worker_interface" "io" "io/ioutil" "log" diff --git a/manager/worker_interface/worker_interface.go b/manager/worker_interface/worker_interface.go index daac355b254..a0f73d74fa4 100644 --- a/manager/worker_interface/worker_interface.go +++ b/manager/worker_interface/worker_interface.go @@ -3,8 +3,8 @@ package worker_interface import ( "errors" "fmt" - "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/db" + "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/db" "time" ) diff --git a/mock/api/manager.go b/mock/api/manager.go index 9c90e0e0efd..976dac30344 100644 --- a/mock/api/manager.go +++ b/mock/api/manager.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/kubeflow/hp-tuning/api (interfaces: ManagerClient) +// Source: github.com/kubeflow/katib/api (interfaces: ManagerClient) // Package mock is a generated GoMock package. package mock @@ -7,7 +7,7 @@ package mock import ( context "context" gomock "github.com/golang/mock/gomock" - api "github.com/kubeflow/hp-tuning/api" + api "github.com/kubeflow/katib/api" grpc "google.golang.org/grpc" reflect "reflect" ) diff --git a/mock/api/suggestion.go b/mock/api/suggestion.go index 9a108e0e91c..6756f1e2b8a 100644 --- a/mock/api/suggestion.go +++ b/mock/api/suggestion.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/kubeflow/hp-tuning/api (interfaces: SuggestionClient) +// Source: github.com/kubeflow/katib/api (interfaces: SuggestionClient) // Package mock is a generated GoMock package. package mock @@ -7,7 +7,7 @@ package mock import ( context "context" gomock "github.com/golang/mock/gomock" - api "github.com/kubeflow/hp-tuning/api" + api "github.com/kubeflow/katib/api" grpc "google.golang.org/grpc" reflect "reflect" ) diff --git a/mock/db/db.go b/mock/db/db.go index 09b1efb7c0a..00e19f62edf 100644 --- a/mock/db/db.go +++ b/mock/db/db.go @@ -1,13 +1,13 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/kubeflow/hp-tuning/db (interfaces: VizierDBInterface) +// Source: github.com/kubeflow/katib/db (interfaces: VizierDBInterface) // Package mock is a generated GoMock package. package mock import ( gomock "github.com/golang/mock/gomock" - api "github.com/kubeflow/hp-tuning/api" - db "github.com/kubeflow/hp-tuning/db" + api "github.com/kubeflow/katib/api" + db "github.com/kubeflow/katib/db" reflect "reflect" time "time" ) diff --git a/mock/modelstore/modelstore.go b/mock/modelstore/modelstore.go index 1fe6bfd375b..c5f3e3221a1 100644 --- a/mock/modelstore/modelstore.go +++ b/mock/modelstore/modelstore.go @@ -1,12 +1,12 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/kubeflow/hp-tuning/manager/modelstore (interfaces: ModelStore) +// Source: github.com/kubeflow/katib/manager/modelstore (interfaces: ModelStore) // Package mock is a generated GoMock package. package mock import ( gomock "github.com/golang/mock/gomock" - api "github.com/kubeflow/hp-tuning/api" + api "github.com/kubeflow/katib/api" reflect "reflect" ) diff --git a/mock/worker/worker.go b/mock/worker/worker.go index a8eb2624d33..9ace51e18ac 100644 --- a/mock/worker/worker.go +++ b/mock/worker/worker.go @@ -1,12 +1,12 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/kubeflow/hp-tuning/manager/worker_interface (interfaces: WorkerInterface) +// Source: github.com/kubeflow/katib/manager/worker_interface (interfaces: WorkerInterface) // Package mock is a generated GoMock package. package mock import ( gomock "github.com/golang/mock/gomock" - api "github.com/kubeflow/hp-tuning/api" + api "github.com/kubeflow/katib/api" reflect "reflect" ) diff --git a/prow_config.yaml b/prow_config.yaml index 9470ca0c168..e91effa043c 100644 --- a/prow_config.yaml +++ b/prow_config.yaml @@ -1,6 +1,6 @@ # This file configures the workflows to trigger in our Prow jobs. # see kubeflow/testing/py/run_e2e_workflow.py workflows: - - app_dir: kubeflow/hp-tuning/test/workflows + - app_dir: kubeflow/katib/test/workflows component: workflows name: e2e diff --git a/scripts/mockgen.sh b/scripts/mockgen.sh index fffaa444c02..697d72ad6a0 100755 --- a/scripts/mockgen.sh +++ b/scripts/mockgen.sh @@ -22,13 +22,13 @@ SCRIPT_ROOT=$(dirname ${BASH_SOURCE})/.. cd ${SCRIPT_ROOT} echo "Generating ManagerClient..." -mockgen -package mock -destination mock/api/manager.go github.com/kubeflow/hp-tuning/api ManagerClient +mockgen -package mock -destination mock/api/manager.go github.com/kubeflow/katib/api ManagerClient echo "Generating SuggestionClient..." -mockgen -package mock -destination mock/api/suggestion.go github.com/kubeflow/hp-tuning/api SuggestionClient +mockgen -package mock -destination mock/api/suggestion.go github.com/kubeflow/katib/api SuggestionClient echo "Generating VizierDBInterface..." -mockgen -package mock -destination mock/db/db.go github.com/kubeflow/hp-tuning/db VizierDBInterface +mockgen -package mock -destination mock/db/db.go github.com/kubeflow/katib/db VizierDBInterface echo "Generating WorkerInterface..." -mockgen -package mock -destination mock/worker/worker.go github.com/kubeflow/hp-tuning/manager/worker_interface WorkerInterface +mockgen -package mock -destination mock/worker/worker.go github.com/kubeflow/katib/manager/worker_interface WorkerInterface echo "Generating ModelStore..." -mockgen -package mock -destination mock/modelstore/modelstore.go github.com/kubeflow/hp-tuning/manager/modelstore ModelStore +mockgen -package mock -destination mock/modelstore/modelstore.go github.com/kubeflow/katib/manager/modelstore ModelStore cd - > /dev/null diff --git a/suggestion/bayesianoptimization/Dockerfile b/suggestion/bayesianoptimization/Dockerfile index 5aaf9637ab8..58037727901 100644 --- a/suggestion/bayesianoptimization/Dockerfile +++ b/suggestion/bayesianoptimization/Dockerfile @@ -1,8 +1,8 @@ FROM python:3 -ADD . /usr/src/app/github.com/kubeflow/hp-tuning -WORKDIR /usr/src/app/github.com/kubeflow/hp-tuning/suggestion/bayesianoptimization +ADD . /usr/src/app/github.com/kubeflow/katib +WORKDIR /usr/src/app/github.com/kubeflow/katib/suggestion/bayesianoptimization RUN pip install --no-cache-dir -r requirements.txt -ENV PYTHONPATH /usr/src/app/github.com/kubeflow/hp-tuning +ENV PYTHONPATH /usr/src/app/github.com/kubeflow/katib -ENTRYPOINT ["python", "main.py"] \ No newline at end of file +ENTRYPOINT ["python", "main.py"] diff --git a/suggestion/grid/Dockerfile b/suggestion/grid/Dockerfile index 1fda4746aa7..671e43fed2a 100644 --- a/suggestion/grid/Dockerfile +++ b/suggestion/grid/Dockerfile @@ -1,10 +1,10 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/suggestion/grid +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/suggestion/grid RUN go build -o grid FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/suggestion/grid /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/suggestion/grid /app/ ENTRYPOINT ["./grid"] diff --git a/suggestion/grid/main.go b/suggestion/grid/main.go index 990a4159631..d66452bf05e 100644 --- a/suggestion/grid/main.go +++ b/suggestion/grid/main.go @@ -7,8 +7,8 @@ import ( "google.golang.org/grpc" "google.golang.org/grpc/reflection" - pb "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/suggestion" + pb "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/suggestion" ) func main() { diff --git a/suggestion/grid_service.go b/suggestion/grid_service.go index 204afbb52f1..ff8bed6730d 100644 --- a/suggestion/grid_service.go +++ b/suggestion/grid_service.go @@ -6,7 +6,7 @@ import ( "log" "strconv" - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" ) type GridSuggestParameters struct { diff --git a/suggestion/hyperband/Dockerfile b/suggestion/hyperband/Dockerfile index eee2de9a93c..ae8f415e097 100644 --- a/suggestion/hyperband/Dockerfile +++ b/suggestion/hyperband/Dockerfile @@ -1,10 +1,10 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/suggestion/hyperband +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/suggestion/hyperband RUN go build -o hyperband FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/suggestion/hyperband /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/suggestion/hyperband /app/ ENTRYPOINT ["./hyperband"] diff --git a/suggestion/hyperband/main.go b/suggestion/hyperband/main.go index 4463c5dbfe9..26312dddc43 100644 --- a/suggestion/hyperband/main.go +++ b/suggestion/hyperband/main.go @@ -7,8 +7,8 @@ import ( "google.golang.org/grpc" "google.golang.org/grpc/reflection" - pb "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/suggestion" + pb "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/suggestion" ) func main() { diff --git a/suggestion/hyperband_service.go b/suggestion/hyperband_service.go index cf1c99d141a..607464683a2 100644 --- a/suggestion/hyperband_service.go +++ b/suggestion/hyperband_service.go @@ -4,13 +4,13 @@ import ( "context" "crypto/rand" "fmt" - "github.com/kubeflow/hp-tuning/db" + "github.com/kubeflow/katib/db" "log" "math" "sort" "strconv" - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" ) type Bracket []*api.Trial diff --git a/suggestion/random/Dockerfile b/suggestion/random/Dockerfile index 3fc89518d75..c59c69fd65c 100644 --- a/suggestion/random/Dockerfile +++ b/suggestion/random/Dockerfile @@ -1,10 +1,10 @@ FROM golang:alpine AS build-env # The GOPATH in the image is /go. -ADD . /go/src/github.com/kubeflow/hp-tuning -WORKDIR /go/src/github.com/kubeflow/hp-tuning/suggestion/random +ADD . /go/src/github.com/kubeflow/katib +WORKDIR /go/src/github.com/kubeflow/katib/suggestion/random RUN go build -o random FROM alpine:3.7 WORKDIR /app -COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/suggestion/random /app/ +COPY --from=build-env /go/src/github.com/kubeflow/katib/suggestion/random /app/ ENTRYPOINT ["./random"] diff --git a/suggestion/random/main.go b/suggestion/random/main.go index bddb2e7a4ad..c653b0f0110 100644 --- a/suggestion/random/main.go +++ b/suggestion/random/main.go @@ -7,8 +7,8 @@ import ( "google.golang.org/grpc" "google.golang.org/grpc/reflection" - pb "github.com/kubeflow/hp-tuning/api" - "github.com/kubeflow/hp-tuning/suggestion" + pb "github.com/kubeflow/katib/api" + "github.com/kubeflow/katib/suggestion" ) func main() { diff --git a/suggestion/random_service.go b/suggestion/random_service.go index 2b10bce2daf..c180d13c42e 100644 --- a/suggestion/random_service.go +++ b/suggestion/random_service.go @@ -7,7 +7,7 @@ import ( "strconv" "time" - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" ) type RandomSuggestParameters struct { diff --git a/suggestion/types.go b/suggestion/types.go index 47fc608ee78..0a42ca89d30 100644 --- a/suggestion/types.go +++ b/suggestion/types.go @@ -3,7 +3,7 @@ package suggestion import ( "context" - "github.com/kubeflow/hp-tuning/api" + "github.com/kubeflow/katib/api" ) const ( diff --git a/test/scripts/build.sh b/test/scripts/build.sh index e3ef76606e0..5189c9b0d88 100755 --- a/test/scripts/build.sh +++ b/test/scripts/build.sh @@ -36,13 +36,13 @@ ln -s ${PWD} ${GO_DIR} cd ${GO_DIR} echo "Build operator binary" mkdir bin -go build -o bin/katib-core github.com/kubeflow/hp-tuning/manager -go build -o bin/dlkmanager github.com/kubeflow/hp-tuning//dlk/dlkmanager -go build -o bin/katib-suggestion-grid github.com/kubeflow/hp-tuning/suggestion/grid -go build -o bin/katib-suggestion-hyperband github.com/kubeflow/hp-tuning/suggestion/hyperband -go build -o bin/katib-suggestion-random github.com/kubeflow/hp-tuning/suggestion/random -go build -o bin/katib-earlystopping-medianstopping github.com/kubeflow/hp-tuning/earlystopping/medianstopping -go build -o bin/katib github.com/kubeflow/hp-tuning/cli +go build -o bin/katib-core github.com/kubeflow/katib/manager +go build -o bin/dlkmanager github.com/kubeflow/katib//dlk/dlkmanager +go build -o bin/katib-suggestion-grid github.com/kubeflow/katib/suggestion/grid +go build -o bin/katib-suggestion-hyperband github.com/kubeflow/katib/suggestion/hyperband +go build -o bin/katib-suggestion-random github.com/kubeflow/katib/suggestion/random +go build -o bin/katib-earlystopping-medianstopping github.com/kubeflow/katib/earlystopping/medianstopping +go build -o bin/katib github.com/kubeflow/katib/cli #echo "building container in gcloud" #gcloud version # gcloud components update -q diff --git a/test/workflows/components/workflows.libsonnet b/test/workflows/components/workflows.libsonnet index 3b87b638256..5c9acc77254 100644 --- a/test/workflows/components/workflows.libsonnet +++ b/test/workflows/components/workflows.libsonnet @@ -52,8 +52,8 @@ local goDir = testDir + "/go"; // Source directory where all repos should be checked out local srcRootDir = testDir + "/src"; - // The directory containing the kubeflow/hp-tuning repo - local srcDir = srcRootDir + "/kubeflow/hp-tuning"; + // The directory containing the kubeflow/katib repo + local srcDir = srcRootDir + "/kubeflow/katib"; local testWorkerImage = "gcr.io/mlkube-testing/test-worker"; local golangImage = "golang:1.9.4-stretch"; // TODO(jose5918) Build our own helm image @@ -89,7 +89,7 @@ name; local zone = params.zone; local registry = params.registry; - local chart = srcDir + "/hp-tuning-chart"; + local chart = srcDir + "/katib-chart"; { // Build an Argo template to execute a particular command. // step_name: Name for the template