fix tests

Author: a-r-r-o-w

src/diffusers/guiders/guider_utils.py

@@ -13,15 +13,14 @@
 # limitations under the License.
 
 import os
-from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 import torch
 from huggingface_hub.utils import validate_hf_hub_args
 from typing_extensions import Self
 
 from ..configuration_utils import ConfigMixin
-from ..utils import PushToHubMixin, get_logger
+from ..utils import BaseOutput, PushToHubMixin, get_logger
 
 
 if TYPE_CHECKING:
@@ -285,8 +284,7 @@ def save_pretrained(self, save_directory: Union[str, os.PathLike], push_to_hub:
         self.save_config(save_directory=save_directory, push_to_hub=push_to_hub, **kwargs)
 
 
-@dataclass
-class GuiderOutput:
+class GuiderOutput(BaseOutput):
     pred: torch.Tensor
     pred_cond: Optional[torch.Tensor]
     pred_uncond: Optional[torch.Tensor]