

"""simple docstring"""




from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


a_         =      {
    "configuration_openai": ["OPENAI_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP", "OpenAIGPTConfig"],
    "tokenization_openai": ["OpenAIGPTTokenizer"],
}

try:
			if not is_tokenizers_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			pass
else:
			a_         =      ["OpenAIGPTTokenizerFast"]

try:
			if not is_torch_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			pass
else:
			a_         =      [
			    "OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST",
			    "OpenAIGPTDoubleHeadsModel",
			    "OpenAIGPTForSequenceClassification",
			    "OpenAIGPTLMHeadModel",
			    "OpenAIGPTModel",
			    "OpenAIGPTPreTrainedModel",
			    "load_tf_weights_in_openai_gpt",
			]

try:
			if not is_tf_available():
						raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
			pass
else:
			a_         =      [
			    "TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST",
			    "TFOpenAIGPTDoubleHeadsModel",
			    "TFOpenAIGPTForSequenceClassification",
			    "TFOpenAIGPTLMHeadModel",
			    "TFOpenAIGPTMainLayer",
			    "TFOpenAIGPTModel",
			    "TFOpenAIGPTPreTrainedModel",
			]


if TYPE_CHECKING:
			from .configuration_openai import OPENAI_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP, OpenAIGPTConfig
			from .tokenization_openai import OpenAIGPTTokenizer

			try:
						if not is_tokenizers_available():
									raise OptionalDependencyNotAvailable()
			except OptionalDependencyNotAvailable:
						pass
			else:
						from .tokenization_openai_fast import OpenAIGPTTokenizerFast

			try:
						if not is_torch_available():
									raise OptionalDependencyNotAvailable()
			except OptionalDependencyNotAvailable:
						pass
			else:
						from .modeling_openai import (
						    OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST,
						    OpenAIGPTDoubleHeadsModel,
						    OpenAIGPTForSequenceClassification,
						    OpenAIGPTLMHeadModel,
						    OpenAIGPTModel,
						    OpenAIGPTPreTrainedModel,
						    load_tf_weights_in_openai_gpt,
						)

			try:
						if not is_tf_available():
									raise OptionalDependencyNotAvailable()
			except OptionalDependencyNotAvailable:
						pass
			else:
						from .modeling_tf_openai import (
						    TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST,
						    TFOpenAIGPTDoubleHeadsModel,
						    TFOpenAIGPTForSequenceClassification,
						    TFOpenAIGPTLMHeadModel,
						    TFOpenAIGPTMainLayer,
						    TFOpenAIGPTModel,
						    TFOpenAIGPTPreTrainedModel,
						)

else:
			import sys

			a_         =      _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)


import math
import os

import cva
import numpy as np

a_         =      255.0

def 	lowerCamelCase__	(   _a							,     _a):
	SCREAMING_SNAKE_CASE				: Optional[int]			       =    np.mean((original - contrast) ** 2)
	if mse == 0:
		return 100

	return 20 * math.logaa(PIXEL_MAX / math.sqrt(_a))

def 	lowerCamelCase__	(   ):
	SCREAMING_SNAKE_CASE				: List[Any]			       =    os.path.dirname(os.path.realpath(_a))
	# Loading images (original image and compressed image)
	SCREAMING_SNAKE_CASE				: Any			       =    cva.imread(os.path.join(_a							,     "image_data/original_image.png"))
	SCREAMING_SNAKE_CASE				: List[str]			       =    cva.imread(os.path.join(_a							,     "image_data/compressed_image.png")							,     1)

	SCREAMING_SNAKE_CASE				: List[Any]			       =    cva.imread(os.path.join(_a							,     "image_data/PSNR-example-base.png"))
	SCREAMING_SNAKE_CASE				: List[str]			       =    cva.imread(
	    os.path.join(_a							,     "image_data/PSNR-example-comp-10.jpg")							,     1)

	# Value expected: 29.73dB
	print("-- First Test --")
	print(f"PSNR value is {peak_signal_to_noise_ratio(_a							,     _a)} dB")

	# # Value expected: 31.53dB (Wikipedia Example)
	print("\n-- Second Test --")
	print(f"PSNR value is {peak_signal_to_noise_ratio(_a							,     _a)} dB")


if __name__ == "__main__":
			main()