From b8e64ece0a7b592ea3dec604628515595ef5c9fc Mon Sep 17 00:00:00 2001 From: Alexander Visheratin Date: Sat, 13 Apr 2024 00:16:29 -0400 Subject: [PATCH] CI fixes. --- src/transformers/__init__.py | 2 +- src/transformers/utils/dummy_pt_objects.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index a413c5636d2585..05d08d908b1ad8 100644 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -3855,8 +3855,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 index 3fe961f0f66392..1436671b232dae 100644 --- a/src/transformers/utils/dummy_pt_objects.py +++ b/src/transformers/utils/dummy_pt_objects.py @@ -9880,14 +9880,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"]