Fix hub_retry (#36449)
* cry * trigger --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user