Create `xpk` package with `utils` module

AI-Hypercomputer / xpk

xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.

Apache License 2.0

70 stars 18 forks source link

Create `xpk` package with `utils` module #166

Closed PBundyra closed 1 month ago

PBundyra commented 1 month ago

Fixes / Features

Create xpk package
Create utils module
Move logic from xpk.py to src/xpk/main.py, and leave xpk.py as an entrypoint
Install xpk with pip as part of GH CI/CD workflows to verify it

This PR serves as a preparatory PR for further refactor. It introduces utils.py module to set an example for further changes.

Testing / Documentation

Tested manually

[ y ] Tests pass
[ n ] Appropriate changes to documentation are included in the PR - no user-facing changes

PBundyra commented 1 month ago

cc @sharabiani @44past4

PBundyra commented 1 month ago

I will update and merge once we resolve discussion around the structure and naming

PBundyra commented 1 month ago

@sharabiani @Obliviour PTAL