
import pytest

import datasets


# Import fixture modules as plugins
SCREAMING_SNAKE_CASE				:str									=							['tests.fixtures.files', 'tests.fixtures.hub', 'tests.fixtures.fsspec']


def 		UpperCAmelCase     (      a_						,  a_		)						->      Optional[Any]:







				"""simple docstring"""



				for item in items:
								if any(marker in item.keywords for marker in ["integration", "unit"]		):
												continue
								item.add_marker(pytest.mark.unit		)


def 		UpperCAmelCase     (      a_		)						->      str:







				"""simple docstring"""



				config.addinivalue_line("markers"						,  "torchaudio_latest: mark test to run with torchaudio>=0.12"		)


@pytest.fixture(autouse=a_		)
def 		UpperCAmelCase     (      a_						,  a_		)						->      Optional[int]:







				"""simple docstring"""



				__A       =					tmp_path_factory.getbasetemp() / "cache"
				__A       =					test_hf_cache_home / "datasets"
				__A       =					test_hf_cache_home / "metrics"
				__A       =					test_hf_cache_home / "modules"
				monkeypatch.setattr("datasets.config.HF_DATASETS_CACHE"						,  str(a_		)		)
				monkeypatch.setattr("datasets.config.HF_METRICS_CACHE"						,  str(a_		)		)
				monkeypatch.setattr("datasets.config.HF_MODULES_CACHE"						,  str(a_		)		)
				__A       =					test_hf_datasets_cache / "downloads"
				monkeypatch.setattr("datasets.config.DOWNLOADED_DATASETS_PATH"						,  str(a_		)		)
				__A       =					test_hf_datasets_cache / "downloads" / "extracted"
				monkeypatch.setattr("datasets.config.EXTRACTED_DATASETS_PATH"						,  str(a_		)		)


@pytest.fixture(autouse=a_						,  scope="session"		)
def 		UpperCAmelCase     (      )						->      Union[str, Any]:







				"""simple docstring"""



				datasets.disable_progress_bar()


@pytest.fixture(autouse=a_		)
def 		UpperCAmelCase     (      a_		)						->      Union[str, Any]:







				"""simple docstring"""



				monkeypatch.setattr("datasets.config.HF_UPDATE_DOWNLOAD_COUNTS"						,  a_		)


@pytest.fixture
def 		UpperCAmelCase     (      a_		)						->      str:







				"""simple docstring"""



				monkeypatch.setattr("sqlalchemy.util.deprecations.SILENCE_UBER_WARNING"						,  a_		)





"""simple docstring"""







from ..utils import (
    OptionalDependencyNotAvailable,
    is_flax_available,
    is_scipy_available,
    is_torch_available,
    is_torchsde_available,
)


try:
			if not is_torch_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ..utils.dummy_pt_objects import *  # noqa F403
else:
			from .scheduling_consistency_models import CMStochasticIterativeScheduler
			from .scheduling_ddim import DDIMScheduler
			from .scheduling_ddim_inverse import DDIMInverseScheduler
			from .scheduling_ddim_parallel import DDIMParallelScheduler
			from .scheduling_ddpm import DDPMScheduler
			from .scheduling_ddpm_parallel import DDPMParallelScheduler
			from .scheduling_deis_multistep import DEISMultistepScheduler
			from .scheduling_dpmsolver_multistep import DPMSolverMultistepScheduler
			from .scheduling_dpmsolver_multistep_inverse import DPMSolverMultistepInverseScheduler
			from .scheduling_dpmsolver_singlestep import DPMSolverSinglestepScheduler
			from .scheduling_euler_ancestral_discrete import EulerAncestralDiscreteScheduler
			from .scheduling_euler_discrete import EulerDiscreteScheduler
			from .scheduling_heun_discrete import HeunDiscreteScheduler
			from .scheduling_ipndm import IPNDMScheduler
			from .scheduling_k_dpm_2_ancestral_discrete import KDPMaAncestralDiscreteScheduler
			from .scheduling_k_dpm_2_discrete import KDPMaDiscreteScheduler
			from .scheduling_karras_ve import KarrasVeScheduler
			from .scheduling_pndm import PNDMScheduler
			from .scheduling_repaint import RePaintScheduler
			from .scheduling_sde_ve import ScoreSdeVeScheduler
			from .scheduling_sde_vp import ScoreSdeVpScheduler
			from .scheduling_unclip import UnCLIPScheduler
			from .scheduling_unipc_multistep import UniPCMultistepScheduler
			from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin
			from .scheduling_vq_diffusion import VQDiffusionScheduler

try:
			if not is_flax_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ..utils.dummy_flax_objects import *  # noqa F403
else:
			from .scheduling_ddim_flax import FlaxDDIMScheduler
			from .scheduling_ddpm_flax import FlaxDDPMScheduler
			from .scheduling_dpmsolver_multistep_flax import FlaxDPMSolverMultistepScheduler
			from .scheduling_karras_ve_flax import FlaxKarrasVeScheduler
			from .scheduling_lms_discrete_flax import FlaxLMSDiscreteScheduler
			from .scheduling_pndm_flax import FlaxPNDMScheduler
			from .scheduling_sde_ve_flax import FlaxScoreSdeVeScheduler
			from .scheduling_utils_flax import (
			    FlaxKarrasDiffusionSchedulers,
			    FlaxSchedulerMixin,
			    FlaxSchedulerOutput,
			    broadcast_to_shape_from_left,
			)


try:
			if not (is_torch_available() and is_scipy_available()):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ..utils.dummy_torch_and_scipy_objects import *  # noqa F403
else:
			from .scheduling_lms_discrete import LMSDiscreteScheduler

try:
			if not (is_torch_available() and is_torchsde_available()):
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			from ..utils.dummy_torch_and_torchsde_objects import *  # noqa F403
else:
			from .scheduling_dpmsolver_sde import DPMSolverSDEScheduler




