kubeflow / sdk / 24414443771

83%

push

github

fix(hub): add params to register_model() (#440)

* fix(hub): add storage_key, storage_path, service_account_name to register_model

Signed-off-by: 1Ayush-Petwal <ayushpetwal.0105@gmail.com>

* refactor(hub): group storage args into StorageConfig dataclass

Replace the three flat kwargs (storage_key, storage_path,
service_account_name) on register_model() with a single
storage_config: StorageConfig | None = None parameter.

Introduces kubeflow/hub/types/types.py with the StorageConfig
dataclass (matching the trainer/optimizer/spark types/ pattern),
re-exports it from kubeflow.hub, updates the docstring, and adds
parametrized test cases verifying correct field forwarding.

Signed-off-by: 1Ayush-Petwal <ayushpetwal.0105@gmail.com>

---------

Signed-off-by: 1Ayush-Petwal <ayushpetwal.0105@gmail.com>

20 of 20 new or added lines in 4 files covered. (100.0%)

4565 of 5727 relevant lines covered (79.71%)

1.59 hits per line