CI fixes.

huggingface · Apr 24, 2024 · e8eadd2 · e8eadd2
1 parent 4e1aeef
commit e8eadd2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -3910,8 +3910,8 @@
         "get_inverse_sqrt_schedule",
         "get_linear_schedule_with_warmup",
         "get_polynomial_decay_schedule_with_warmup",
-        "get_wsd_schedule",
         "get_scheduler",
+        "get_wsd_schedule",
     ]
     _import_structure["pytorch_utils"] = [
         "Conv1D",

diff --git a/src/transformers/utils/dummy_pt_objects.py b/src/transformers/utils/dummy_pt_objects.py
@@ -10019,14 +10019,14 @@ def get_polynomial_decay_schedule_with_warmup(*args, **kwargs):
     requires_backends(get_polynomial_decay_schedule_with_warmup, ["torch"])
 
 
-def get_wsd_schedule(*args, **kwargs):
-    requires_backends(get_wsd_schedule, ["torch"])
-
-
 def get_scheduler(*args, **kwargs):
     requires_backends(get_scheduler, ["torch"])
 
 
+def get_wsd_schedule(*args, **kwargs):
+    requires_backends(get_wsd_schedule, ["torch"])
+
+
 class Conv1D(metaclass=DummyObject):
     _backends = ["torch"]