kubeflow / sdk / 16727070395

80%

Build Type

push

github

Committed by web-flow

Commit Message

feat(trainer): Support Framework Labels in Runtimes (#56)

* Refactor SDK to support runtime labels

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

* Fix unit tests

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

* Remove legacy types

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

* Add warnings if runtime doesn't have a label

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

* Change to TORCH_TUNE

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

* Add validation for runtime trainers

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

* Print Runtime name

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

* Fix tests

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

---------

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

Run Details

48 of 84 new or added lines in 2 files covered. (57.14%)

1 existing line in 1 file now uncovered.

280 of 424 relevant lines covered (66.04%)

0.66 hits per line