Fix hub_retry (#36449)

* 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

View File

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