






import re
import string

import numpy as np

import datasets


UpperCAmelCase_ :						Dict      =      '''
Returns the rate at which the input predicted strings exactly match their references, ignoring any strings input as part of the regexes_to_ignore list.
'''

UpperCAmelCase_ :						Any      =      '''
Args:
    predictions: List of predicted texts.
    references: List of reference texts.
    regexes_to_ignore: List, defaults to None. Regex expressions of characters to
        ignore when calculating the exact matches. Note: these regexes are removed
        from the input data before the changes based on the options below (e.g. ignore_case,
        ignore_punctuation, ignore_numbers) are applied.
    ignore_case: Boolean, defaults to False. If true, turns everything
        to lowercase so that capitalization differences are ignored.
    ignore_punctuation: Boolean, defaults to False. If true, removes all punctuation before
        comparing predictions and references.
    ignore_numbers: Boolean, defaults to False. If true, removes all punctuation before
        comparing predictions and references.
Returns:
    exact_match: Dictionary containing exact_match rate. Possible values are between 0.0 and 100.0, inclusive.
Examples:
    >>> exact_match = datasets.load_metric("exact_match")
    >>> refs = ["the cat", "theater", "YELLING", "agent007"]
    >>> preds = ["cat?", "theater", "yelling", "agent"]
    >>> results = exact_match.compute(references=refs, predictions=preds)
    >>> print(round(results["exact_match"], 1))
    25.0

    >>> exact_match = datasets.load_metric("exact_match")
    >>> refs = ["the cat", "theater", "YELLING", "agent007"]
    >>> preds = ["cat?", "theater", "yelling", "agent"]
    >>> results = exact_match.compute(references=refs, predictions=preds, regexes_to_ignore=["the ", "yell"], ignore_case=True, ignore_punctuation=True)
    >>> print(round(results["exact_match"], 1))
    50.0


    >>> exact_match = datasets.load_metric("exact_match")
    >>> refs = ["the cat", "theater", "YELLING", "agent007"]
    >>> preds = ["cat?", "theater", "yelling", "agent"]
    >>> results = exact_match.compute(references=refs, predictions=preds, regexes_to_ignore=["the ", "yell", "YELL"], ignore_case=True, ignore_punctuation=True)
    >>> print(round(results["exact_match"], 1))
    75.0

    >>> exact_match = datasets.load_metric("exact_match")
    >>> refs = ["the cat", "theater", "YELLING", "agent007"]
    >>> preds = ["cat?", "theater", "yelling", "agent"]
    >>> results = exact_match.compute(references=refs, predictions=preds, regexes_to_ignore=["the ", "yell", "YELL"], ignore_case=True, ignore_punctuation=True, ignore_numbers=True)
    >>> print(round(results["exact_match"], 1))
    100.0

    >>> exact_match = datasets.load_metric("exact_match")
    >>> refs = ["The cat sat on the mat.", "Theaters are great.", "It\'s like comparing oranges and apples."]
    >>> preds = ["The cat sat on the mat?", "Theaters are great.", "It\'s like comparing apples and oranges."]
    >>> results = exact_match.compute(references=refs, predictions=preds)
    >>> print(round(results["exact_match"], 1))
    33.3

'''

UpperCAmelCase_ :						Tuple      =      '''
'''

@datasets.utils.file_utils.add_start_docstrings(_DESCRIPTION		,   _KWARGS_DESCRIPTION      )
class     _SCREAMING_SNAKE_CASE       (			datasets.Metric      ):





	def 						_A (		self				:					Optional[int]	):
		return datasets.MetricInfo(
		    description=_DESCRIPTION ,  citation=_CITATION ,  inputs_description=_KWARGS_DESCRIPTION ,  features=datasets.Features(
		        {
		            """predictions""": datasets.Value("""string""" ,  id="""sequence"""	),
		            """references""": datasets.Value("""string""" ,  id="""sequence"""	),
		        }	) ,  reference_urls=[] ,  )





	def 						_A (		self				:					Optional[int] ,  __lowerCamelCase				:					Union[str, Any] ,  __lowerCamelCase				:					List[str] ,  __lowerCamelCase				:					List[str]=None ,  __lowerCamelCase				:					Optional[int]=False ,  __lowerCamelCase				:					int=False ,  __lowerCamelCase				:					Optional[int]=False ,  ):
		if regexes_to_ignore is not None:
			for s in regexes_to_ignore:
				UpperCamelCase     :str          =	np.array([re.sub(__lowerCamelCase ,  """""" ,  __lowerCamelCase	) for x in predictions]	)
				UpperCamelCase     :Tuple          =	np.array([re.sub(__lowerCamelCase ,  """""" ,  __lowerCamelCase	) for x in references]	)
		else:
			UpperCamelCase     :Any          =	np.asarray(__lowerCamelCase	)
			UpperCamelCase     :str          =	np.asarray(__lowerCamelCase	)

		if ignore_case:
			UpperCamelCase     :Tuple          =	np.char.lower(__lowerCamelCase	)
			UpperCamelCase     :Any          =	np.char.lower(__lowerCamelCase	)

		if ignore_punctuation:
			UpperCamelCase     :Optional[int]          =	string.punctuation.maketrans("""""" ,  """""" ,  string.punctuation	)
			UpperCamelCase     :Optional[Any]          =	np.char.translate(__lowerCamelCase ,  table=__lowerCamelCase	)
			UpperCamelCase     :List[str]          =	np.char.translate(__lowerCamelCase ,  table=__lowerCamelCase	)

		if ignore_numbers:
			UpperCamelCase     :Tuple          =	string.digits.maketrans("""""" ,  """""" ,  string.digits	)
			UpperCamelCase     :Dict          =	np.char.translate(__lowerCamelCase ,  table=__lowerCamelCase	)
			UpperCamelCase     :Tuple          =	np.char.translate(__lowerCamelCase ,  table=__lowerCamelCase	)

		UpperCamelCase     :int          =	predictions == references

		return {"exact_match": np.mean(__lowerCamelCase	) * 100}







import torch

from diffusers import DDIMParallelScheduler

from .test_schedulers import SchedulerCommonTest




class     SCREAMING_SNAKE_CASE__ (						UpperCamelCase__      ):
 __SCREAMING_SNAKE_CASE							       =	(DDIMParallelScheduler,)
 __SCREAMING_SNAKE_CASE							       =	(('''eta''', 0.0), ('''num_inference_steps''', 50))



 def 				UpperCamelCase			(							self,**__lowerCamelCase ):
        A__											=						{
            '''num_train_timesteps''': 1000,
            '''beta_start''': 0.0001,
            '''beta_end''': 0.02,
            '''beta_schedule''': '''linear''',
            '''clip_sample''': True,
        }

        config.update(**__lowerCamelCase )
        return config



 def 				UpperCamelCase			(							self,**__lowerCamelCase ):
        A__											=						self.scheduler_classes[0]
        A__											=						self.get_scheduler_config(**__lowerCamelCase )
        A__											=						scheduler_class(**__lowerCamelCase )

        A__					,	A__											=						10, 0.0

        A__											=						self.dummy_model()
        A__											=						self.dummy_sample_deter

        scheduler.set_timesteps(__lowerCamelCase )

        for t in scheduler.timesteps:
               A__											=						model(__lowerCamelCase,__lowerCamelCase )
               A__											=						scheduler.step(__lowerCamelCase,__lowerCamelCase,__lowerCamelCase,__lowerCamelCase ).prev_sample

        return sample



 def 				UpperCamelCase			(							self ):
        for timesteps in [100, 500, 1000]:
               self.check_over_configs(num_train_timesteps=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for steps_offset in [0, 1]:
               self.check_over_configs(steps_offset=__lowerCamelCase )

        A__											=						self.scheduler_classes[0]
        A__											=						self.get_scheduler_config(steps_offset=1 )
        A__											=						scheduler_class(**__lowerCamelCase )
        scheduler.set_timesteps(5 )
        assert torch.equal(scheduler.timesteps,torch.LongTensor([801, 601, 401, 201, 1] ) )



 def 				UpperCamelCase			(							self ):
        for beta_start, beta_end in zip([0.0001, 0.001, 0.01, 0.1],[0.002, 0.02, 0.2, 2] ):
               self.check_over_configs(beta_start=__lowerCamelCase,beta_end=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for schedule in ["linear", "squaredcos_cap_v2"]:
               self.check_over_configs(beta_schedule=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for prediction_type in ["epsilon", "v_prediction"]:
               self.check_over_configs(prediction_type=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for clip_sample in [True, False]:
               self.check_over_configs(clip_sample=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for timestep_spacing in ["trailing", "leading"]:
               self.check_over_configs(timestep_spacing=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for rescale_betas_zero_snr in [True, False]:
               self.check_over_configs(rescale_betas_zero_snr=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        self.check_over_configs(thresholding=__lowerCamelCase )
        for threshold in [0.5, 1.0, 2.0]:
               for prediction_type in ["epsilon", "v_prediction"]:
                      self.check_over_configs(
                          thresholding=__lowerCamelCase,prediction_type=__lowerCamelCase,sample_max_value=__lowerCamelCase,)



 def 				UpperCamelCase			(							self ):
        for t in [1, 10, 49]:
               self.check_over_forward(time_step=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for t, num_inference_steps in zip([1, 10, 50],[10, 50, 500] ):
               self.check_over_forward(time_step=__lowerCamelCase,num_inference_steps=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        for t, eta in zip([1, 10, 49],[0.0, 0.5, 1.0] ):
               self.check_over_forward(time_step=__lowerCamelCase,eta=__lowerCamelCase )



 def 				UpperCamelCase			(							self ):
        A__											=						self.scheduler_classes[0]
        A__											=						self.get_scheduler_config()
        A__											=						scheduler_class(**__lowerCamelCase )

        assert torch.sum(torch.abs(scheduler._get_variance(0,0 ) - 0.0 ) ) < 1E-5
        assert torch.sum(torch.abs(scheduler._get_variance(420,400 ) - 0.14771 ) ) < 1E-5
        assert torch.sum(torch.abs(scheduler._get_variance(980,960 ) - 0.32460 ) ) < 1E-5
        assert torch.sum(torch.abs(scheduler._get_variance(0,0 ) - 0.0 ) ) < 1E-5
        assert torch.sum(torch.abs(scheduler._get_variance(487,486 ) - 0.00979 ) ) < 1E-5
        assert torch.sum(torch.abs(scheduler._get_variance(999,998 ) - 0.02 ) ) < 1E-5



 def 				UpperCamelCase			(							self ):
        A__											=						self.scheduler_classes[0]
        A__											=						self.get_scheduler_config()
        A__											=						scheduler_class(**__lowerCamelCase )

        A__					,	A__											=						10, 0.0
        scheduler.set_timesteps(__lowerCamelCase )

        A__											=						self.dummy_model()
        A__											=						self.dummy_sample_deter
        A__											=						self.dummy_sample_deter + 0.1
        A__											=						self.dummy_sample_deter - 0.1

        A__											=						samplea.shape[0]
        A__											=						torch.stack([samplea, samplea, samplea],dim=0 )
        A__											=						torch.arange(__lowerCamelCase )[0:3, None].repeat(1,__lowerCamelCase )

        A__											=						model(samples.flatten(0,1 ),timesteps.flatten(0,1 ) )
        A__											=						scheduler.batch_step_no_noise(__lowerCamelCase,timesteps.flatten(0,1 ),samples.flatten(0,1 ),__lowerCamelCase )

        A__											=						torch.sum(torch.abs(__lowerCamelCase ) )
        A__											=						torch.mean(torch.abs(__lowerCamelCase ) )

        assert abs(result_sum.item() - 1147.7904 ) < 1E-2
        assert abs(result_mean.item() - 0.4982 ) < 1E-3



 def 				UpperCamelCase			(							self ):
        A__											=						self.full_loop()

        A__											=						torch.sum(torch.abs(__lowerCamelCase ) )
        A__											=						torch.mean(torch.abs(__lowerCamelCase ) )

        assert abs(result_sum.item() - 172.0067 ) < 1E-2
        assert abs(result_mean.item() - 0.223967 ) < 1E-3



 def 				UpperCamelCase			(							self ):
        A__											=						self.full_loop(prediction_type='''v_prediction''' )

        A__											=						torch.sum(torch.abs(__lowerCamelCase ) )
        A__											=						torch.mean(torch.abs(__lowerCamelCase ) )

        assert abs(result_sum.item() - 52.5302 ) < 1E-2
        assert abs(result_mean.item() - 0.0684 ) < 1E-3



 def 				UpperCamelCase			(							self ):
        # We specify different beta, so that the first alpha is 0.99
        A__											=						self.full_loop(set_alpha_to_one=__lowerCamelCase,beta_start=0.01 )
        A__											=						torch.sum(torch.abs(__lowerCamelCase ) )
        A__											=						torch.mean(torch.abs(__lowerCamelCase ) )

        assert abs(result_sum.item() - 149.8295 ) < 1E-2
        assert abs(result_mean.item() - 0.1951 ) < 1E-3




 def 				UpperCamelCase			(							self ):
        # We specify different beta, so that the first alpha is 0.99
        A__											=						self.full_loop(set_alpha_to_one=__lowerCamelCase,beta_start=0.01 )
        A__											=						torch.sum(torch.abs(__lowerCamelCase ) )
        A__											=						torch.mean(torch.abs(__lowerCamelCase ) )

        assert abs(result_sum.item() - 149.0784 ) < 1E-2
        assert abs(result_mean.item() - 0.1941 ) < 1E-3





