"""simple docstring"""




from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_a   :	Tuple 					=     {'configuration_wavlm': ['WAVLM_PRETRAINED_CONFIG_ARCHIVE_MAP', 'WavLMConfig']}

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     _a   :	str 					=     [
         'WAVLM_PRETRAINED_MODEL_ARCHIVE_LIST',
         'WavLMForAudioFrameClassification',
         'WavLMForCTC',
         'WavLMForSequenceClassification',
         'WavLMForXVector',
         'WavLMModel',
         'WavLMPreTrainedModel',
     ]

if TYPE_CHECKING:
     from .configuration_wavlm import WAVLM_PRETRAINED_CONFIG_ARCHIVE_MAP, WavLMConfig

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_wavlm import (
              WAVLM_PRETRAINED_MODEL_ARCHIVE_LIST,
              WavLMForAudioFrameClassification,
              WavLMForCTC,
              WavLMForSequenceClassification,
              WavLMForXVector,
              WavLMModel,
              WavLMPreTrainedModel,
          )

else:
     import sys

     _a   :	Any 					=     _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)





"""simple docstring"""




import pickle
import unittest

import torch

from accelerate import Accelerator
from accelerate.state import AcceleratorState
from accelerate.test_utils import require_cpu
@require_cpu
class 		__A							(				unittest.TestCase	):






     def 					__A						(	self						):
       _lowerCAmelCase		: str						    =  torch.nn.Linear(10       ,  10						)
       _lowerCAmelCase		: Optional[Any]						    =  torch.optim.SGD(model.parameters()       ,  0.1						)
       _lowerCAmelCase		: Optional[Any]						    =  Accelerator()
       _lowerCAmelCase		: Tuple						    =  accelerator.prepare(a__						)
       try:
         pickle.loads(pickle.dumps(a__						)						)
       except Exception as e:
         self.fail(F"Accelerated optimizer pickling failed with {e}"						)
       AcceleratorState._reset_state()




