kubeflow / sdk / 16655049250

push

github

feat(trainer): Add `wait_for_job_status()` API (#52)

* feat(trainer): Add wait_for_job_status() API

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Fix unit tests

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Add tests for wait_for_job_status

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Update test case for timeout

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Update python/kubeflow/trainer/api/trainer_client.py

Co-authored-by: Anya Kramar <akramar@redhat.com>
Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Add Created status after TrainJob is created

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Wait for Created status

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Use watch API

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Remove time import

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

---------

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>
Co-authored-by: Anya Kramar <akramar@redhat.com>

28 of 34 new or added lines in 2 files covered. (82.35%)

264 of 403 relevant lines covered (65.51%)

0.66 hits per line