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

Fix hub_retry (#36449)

Browse Source 
* cry

* trigger

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-02-27 14:38:25 +01:00
committed by GitHub
parent 6a876462c3
commit 482d17be60
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_common.py
View File

@@ -223,7 +223,7 @@ class ModelTesterMixin:
if attr_name.startswith("test_"): if attr_name.startswith("test_"):
attr = getattr(cls, attr_name) attr = getattr(cls, attr_name)
if callable(attr): if callable(attr):
setattr(cls, attr_name, hub_retry(attr)) setattr(cls, attr_name, hub_retry()(attr))
@property @property
def all_generative_model_classes(self): def all_generative_model_classes(self):