From 482d17be60d1836a7e787c5c3d69d4c3ed171ebc Mon Sep 17 00:00:00 2001
From: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Date: Thu, 27 Feb 2025 14:38:25 +0100
Subject: [PATCH] Fix `hub_retry` (#36449)

* cry

* trigger

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
---
 tests/test_modeling_common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
index 8de8b0584c..bb6f14ce3d 100755
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -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):