kubeflow / sdk / 29089835096

83%

push

github

fix(trainer): validate HuggingFace storage_uri has both user and repo (#575)

The HuggingFace initializers only rejected a storage_uri with an empty
path, so URIs missing either component, such as hf://user/ (no repo) or
hf:///model (no user), were accepted and failed later at download time.

Consolidate the validation into a shared _validate_hf_storage_uri helper
used by both HuggingFaceModelInitializer and HuggingFaceDatasetInitializer,
requiring a non-empty user and repo. Add tests for the dataset initializer,
which had no prior coverage, and for the new rejected forms.

Follow-up to #564.

Signed-off-by: Sahil Kumar Singh <sahilkumargreat12@gmail.com>

19 of 19 new or added lines in 2 files covered. (100.0%)

5075 of 6086 relevant lines covered (83.39%)

1.67 hits per line