SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Mention model_info.id instead of model_info.modelId (#32106)

Browse Source
This commit is contained in:
Lucain
2024-07-22 15:14:47 +02:00
committed by GitHub
parent 0fcfc5ccc9
commit f2a1e3ca68
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/marian/test_modeling_marian.py
View File

@@ -409,7 +409,7 @@ class ModelManagementTests(unittest.TestCase):
@require_torch
def test_model_names(self):
model_list = list_models()
model_ids = [x.modelId for x in model_list if x.modelId.startswith(ORG_NAME)]
model_ids = [x.id for x in model_list if x.id.startswith(ORG_NAME)]
bad_model_ids = [mid for mid in model_ids if "+" in model_ids]
self.assertListEqual([], bad_model_ids)
self.assertGreater(len(model_ids), 500)