



"""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









"""simple docstring"""







import unittest

from transformers.utils.backbone_utils import (
    BackboneMixin,
    get_aligned_output_features_output_indices,
    verify_out_features_out_indices,
)






class       UpperCamelCase__			(   unittest.TestCase      ):

			"""simple docstring"""


			def   snake_case__	(      self		)      ->	Dict:
				A__											=					["a", "b", "c"]

				# Defaults to last layer if both are None
				A__							,							A__											=					get_aligned_output_features_output_indices(SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	["c"]		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	[2]		)

				# Out indices set to match out features
				A__							,							A__											=					get_aligned_output_features_output_indices(["a", "c"]						,	SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	["a", "c"]		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	[0, 2]		)

				# Out features set to match out indices
				A__							,							A__											=					get_aligned_output_features_output_indices(SCREAMING_SNAKE_CASE__						,	[0, 2]						,	SCREAMING_SNAKE_CASE__		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	["a", "c"]		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	[0, 2]		)

				# Out features selected from negative indices
				A__							,							A__											=					get_aligned_output_features_output_indices(SCREAMING_SNAKE_CASE__						,	[-3, -1]						,	SCREAMING_SNAKE_CASE__		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	["a", "c"]		)
				self.assertEqual(SCREAMING_SNAKE_CASE__						,	[-3, -1]		)


			def   snake_case__	(      self		)      ->	Dict:
				# Stage names must be set
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(["a", "b"]						,	(0, 1)						,	SCREAMING_SNAKE_CASE__		)

				# Out features must be a list
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(("a", "b")						,	(0, 1)						,	["a", "b"]		)

				# Out features must be a subset of stage names
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(["a", "b"]						,	(0, 1)						,	["a"]		)

				# Out indices must be a list or tuple
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(SCREAMING_SNAKE_CASE__						,	0						,	["a", "b"]		)

				# Out indices must be a subset of stage names
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(SCREAMING_SNAKE_CASE__						,	(0, 1)						,	["a"]		)

				# Out features and out indices must be the same length
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(["a", "b"]						,	(0,)						,	["a", "b", "c"]		)

				# Out features should match out indices
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(["a", "b"]						,	(0, 2)						,	["a", "b", "c"]		)

				# Out features and out indices should be in order
				with self.assertRaises(SCREAMING_SNAKE_CASE__		):
					verify_out_features_out_indices(["b", "a"]						,	(0, 1)						,	["a", "b"]		)

				# Check passes with valid inputs
				verify_out_features_out_indices(["a", "b", "d"]						,	(0, 1, -1)						,	["a", "b", "c", "d"]		)







			def   snake_case__	(      self		)      ->	List[Any]:
				A__											=					BackboneMixin()

				A__											=					["a", "b", "c"]
				A__											=					["a", "c"]
				A__											=					[0, 2]

				# Check that the output features and indices are set correctly
				self.assertEqual(backbone.out_features						,	["a", "c"]		)
				self.assertEqual(backbone.out_indices						,	[0, 2]		)

				# Check out features and indices are updated correctly
				A__											=					["a", "b"]
				self.assertEqual(backbone.out_features						,	["a", "b"]		)
				self.assertEqual(backbone.out_indices						,	[0, 1]		)

				A__											=					[-3, -1]
				self.assertEqual(backbone.out_features						,	["a", "c"]		)
				self.assertEqual(backbone.out_indices						,	[-3, -1]		)




