Skip to content

Commit 780041f

Browse files
fix: Correct component property keys for Hugging Face in fieldspecs
1 parent 8f380e7 commit 780041f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

internal/metadata/fieldspecs.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ const (
3434
ComponentPropertiesHuggingFaceLibraryName Key = "BOM.metadata.component.properties.huggingface:libraryName"
3535
ComponentPropertiesHuggingFaceDownloads Key = "BOM.metadata.component.properties.huggingface:downloads"
3636
ComponentPropertiesHuggingFaceLikes Key = "BOM.metadata.component.properties.huggingface:likes"
37-
ComponentPropertiesModelCardBaseModel Key = "BOM.metadata.component.properties.huggingface:baseModel"
38-
ComponentPropertiesModelCardContact Key = "BOM.metadata.component.properties.huggingface:modelCardContact"
37+
ComponentPropertiesHuggingFaceBaseModel Key = "BOM.metadata.component.properties.huggingface:baseModel"
38+
ComponentPropertiesHuggingFaceContact Key = "BOM.metadata.component.properties.huggingface:modelCardContact"
3939

4040
// BOM.metadata.component.modelCard.* (MODEL CARD)
4141
ModelCardModelParametersTask Key = "BOM.metadata.component.modelCard.modelParameters.task"
@@ -527,15 +527,15 @@ func Registry() []FieldSpec {
527527
}
528528
return r.Likes, true
529529
}),
530-
hfProp(ComponentPropertiesModelCardBaseModel, 0.2, func(src Source) (any, bool) {
530+
hfProp(ComponentPropertiesHuggingFaceBaseModel, 0.2, func(src Source) (any, bool) {
531531
r := src.Readme
532532
if r == nil {
533533
return nil, false
534534
}
535535
s := strings.TrimSpace(r.BaseModel)
536536
return s, s != ""
537537
}),
538-
hfProp(ComponentPropertiesModelCardContact, 0.2, func(src Source) (any, bool) {
538+
hfProp(ComponentPropertiesHuggingFaceContact, 0.2, func(src Source) (any, bool) {
539539
r := src.Readme
540540
if r == nil {
541541
return nil, false

0 commit comments

Comments
 (0)