



import math_equivalence  # From: git+https://github.com/hendrycks/math.git

import datasets


SCREAMING_SNAKE_CASE_				= "\\n@article{hendrycksmath2021,\n  title={Measuring Mathematical Problem Solving With the MATH Dataset},\n  author={Dan Hendrycks\n    and Collin Burns\n    and Saurav Kadavath\n    and Akul Arora\n    and Steven Basart\n    and Eric Tang\n    and Dawn Song\n    and Jacob Steinhardt},\n  journal={arXiv preprint arXiv:2103.03874},\n  year={2021}\n}\n"


SCREAMING_SNAKE_CASE_				= "\\nThis metric is used to assess performance on the Mathematics Aptitude Test of Heuristics (MATH) dataset.\nIt first canonicalizes the inputs (e.g., converting \"1/2\" to \"\\frac{1}{2}\") and then computes accuracy.\n"


SCREAMING_SNAKE_CASE_				= r"\nCalculates accuracy after canonicalizing inputs.\n\nArgs:\n    predictions: list of predictions to score. Each prediction\n        is a string that contains natural language and LaTex.\n    references: list of reference for each prediction. Each\n        reference is a string that contains natural language\n        and LaTex.\nReturns:\n    accuracy: accuracy after canonicalizing inputs\n        (e.g., converting \"1/2\" to \"\\frac{1}{2}\")\n\nExamples:\n    >>> metric = datasets.load_metric(\"competition_math\")\n    >>> results = metric.compute(references=[\"\\frac{1}{2}\"], predictions=[\"1/2\"])\n    >>> print(results)\n    {'accuracy': 1.0}\n"




@datasets.utils.file_utils.add_end_docstrings(_DESCRIPTION     ,_KWARGS_DESCRIPTION				)
class 			snake_case_						(						datasets.Metric				):






							def snake_case_  (   self      ):
									return datasets.MetricInfo(
									    description=_DESCRIPTION			,     citation=_CITATION			,     inputs_description=_KWARGS_DESCRIPTION			,     features=datasets.Features(
									        {
									            "predictions": datasets.Value("string"      ),
									            "references": datasets.Value("string"      ),
									        }      )			,     homepage="https://github.com/hendrycks/math"			,     codebase_urls=["https://github.com/hendrycks/math"]			,     )






							def snake_case_  (   self			,     a_			,     a_      ):
									a_     :      List[str] 						=     0.0
									for i, j in zip(lowercase_			,     lowercase_      ):
											n_correct += 1.0 if math_equivalence.is_equiv(lowercase_			,     lowercase_      ) else 0.0
									a_     :      Union[str, Any] 						=     n_correct / len(lowercase_      )
									return {
									    "accuracy": accuracy,
									}




"""simple docstring"""
import argparse
import os

import torch

from transformers import (
    XLNetConfig,
    XLNetForQuestionAnswering,
    XLNetForSequenceClassification,
    XLNetLMHeadModel,
    load_tf_weights_in_xlnet,
)
from transformers.utils import CONFIG_NAME, WEIGHTS_NAME, logging


SCREAMING_SNAKE_CASE_				= {
    """cola""": 2,
    """mnli""": 3,
    """mrpc""": 2,
    """sst-2""": 2,
    """sts-b""": 1,
    """qqp""": 2,
    """qnli""": 2,
    """rte""": 2,
    """wnli""": 2,
}


logging.set_verbosity_info()



def        lowerCAmelCase_						(							SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__=None	)			->      int:
		# Initialise PyTorch model
		a_     :      Optional[int] 						=     XLNetConfig.from_json_file(SCREAMING_SNAKE_CASE__	)

		a_     :      List[Any] 						=     finetuning_task.lower() if finetuning_task is not None else ""
		if finetuning_task in GLUE_TASKS_NUM_LABELS:
				print(F"""Building PyTorch XLNetForSequenceClassification model from configuration: {config}"""	)
				a_     :      Tuple 						=     finetuning_task
				a_     :      Dict 						=     GLUE_TASKS_NUM_LABELS[finetuning_task]
				a_     :      Tuple 						=     XLNetForSequenceClassification(SCREAMING_SNAKE_CASE__	)
		elif "squad" in finetuning_task:
				a_     :      Tuple 						=     finetuning_task
				a_     :      List[Any] 						=     XLNetForQuestionAnswering(SCREAMING_SNAKE_CASE__	)
		else:
				a_     :      Any 						=     XLNetLMHeadModel(SCREAMING_SNAKE_CASE__	)

		# Load weights from tf checkpoint
		load_tf_weights_in_xlnet(SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__	)

		# Save pytorch-model
		a_     :      Tuple 						=     os.path.join(SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__	)
		a_     :      str 						=     os.path.join(SCREAMING_SNAKE_CASE__, SCREAMING_SNAKE_CASE__	)
		print(F"""Save PyTorch model to {os.path.abspath(SCREAMING_SNAKE_CASE__	)}"""	)
		torch.save(model.state_dict(), SCREAMING_SNAKE_CASE__	)
		print(F"""Save configuration file to {os.path.abspath(SCREAMING_SNAKE_CASE__	)}"""	)
		with open(SCREAMING_SNAKE_CASE__, "w", encoding="utf-8"	) as f:
				f.write(config.to_json_string()	)


if __name__ == "__main__":
					SCREAMING_SNAKE_CASE_				= argparse.ArgumentParser()
					# Required parameters
					parser.add_argument(
					    """--tf_checkpoint_path""", default=None, type=str, required=True, help="""Path to the TensorFlow checkpoint path."""
					)
					parser.add_argument(
					    """--xlnet_config_file""",
					    default=None,
					    type=str,
					    required=True,
					    help=(
					        """The config json file corresponding to the pre-trained XLNet model. \n"""
					        """This specifies the model architecture."""
					    ),
					)
					parser.add_argument(
					    """--pytorch_dump_folder_path""",
					    default=None,
					    type=str,
					    required=True,
					    help="""Path to the folder to store the PyTorch model or dataset/vocab.""",
					)
					parser.add_argument(
					    """--finetuning_task""",
					    default=None,
					    type=str,
					    help="""Name of a task on which the XLNet TensorFlow model was fine-tuned""",
					)
					SCREAMING_SNAKE_CASE_				= parser.parse_args()
					print(args)

					convert_xlnet_checkpoint_to_pytorch(
					    args.tf_checkpoint_path, args.xlnet_config_file, args.pytorch_dump_folder_path, args.finetuning_task
					)