





import re
import tempfile
from pathlib import Path

import pytest
import yaml

from datasets.utils.readme import ReadMe


# @pytest.fixture
# def example_yaml_structure():

A_												=     yaml.safe_load(
    "\\nname: \"\"\nallow_empty: false\nallow_empty_text: true\nsubsections:\n  - name: \"Dataset Card for X\" # First-level markdown heading\n    allow_empty: false\n    allow_empty_text: true\n    subsections:\n      - name: \"Table of Contents\"\n        allow_empty: false\n        allow_empty_text: false\n        subsections: null\n      - name: \"Dataset Description\"\n        allow_empty: false\n        allow_empty_text: false\n        subsections:\n          - name: \"Dataset Summary\"\n            allow_empty: false\n            allow_empty_text: false\n            subsections: null\n          - name: \"Supported Tasks and Leaderboards\"\n            allow_empty: true\n            allow_empty_text: true\n            subsections: null\n          - name: Languages\n            allow_empty: false\n            allow_empty_text: true\n            subsections: null\n"
)


A_												=     {
    "name": "root",
    "text": "",
    "is_empty_text": True,
    "subsections": [
        {
            "name": "Dataset Card for My Dataset",
            "text": "",
            "is_empty_text": True,
            "subsections": [
                {"name": "Table of Contents", "text": "Some text here.", "is_empty_text": False, "subsections": []},
                {
                    "name": "Dataset Description",
                    "text": "Some text here.",
                    "is_empty_text": False,
                    "subsections": [
                        {
                            "name": "Dataset Summary",
                            "text": "Some text here.",
                            "is_empty_text": False,
                            "subsections": [],
                        },
                        {
                            "name": "Supported Tasks and Leaderboards",
                            "text": "",
                            "is_empty_text": True,
                            "subsections": [],
                        },
                        {"name": "Languages", "text": "Language Text", "is_empty_text": False, "subsections": []},
                    ],
                },
            ],
        }
    ],
}


A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"


A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n#### Extra Ignored Subsection\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

A_												=     {
    "name": "root",
    "text": "",
    "is_empty_text": True,
    "subsections": [
        {
            "name": "Dataset Card for My Dataset",
            "text": "",
            "is_empty_text": True,
            "subsections": [
                {"name": "Table of Contents", "text": "Some text here.", "is_empty_text": False, "subsections": []},
                {
                    "name": "Dataset Description",
                    "text": "Some text here.",
                    "is_empty_text": False,
                    "subsections": [
                        {
                            "name": "Dataset Summary",
                            "text": "Some text here.",
                            "is_empty_text": False,
                            "subsections": [
                                {
                                    "name": "Extra Ignored Subsection",
                                    "text": "",
                                    "is_empty_text": True,
                                    "subsections": [],
                                }
                            ],
                        },
                        {
                            "name": "Supported Tasks and Leaderboards",
                            "text": "",
                            "is_empty_text": True,
                            "subsections": [],
                        },
                        {"name": "Languages", "text": "Language Text", "is_empty_text": False, "subsections": []},
                    ],
                },
            ],
        }
    ],
}

A_												=     "\\n---\n---\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

A_												=     (
    "The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README."
)

A_												=     "\\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

A_												=     (
    "The following issues were found for the README at `{path}`:\n-\tNo YAML markers are present in the README."
)

A_												=     "\\n---\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

A_												=     "The following issues were found for the README at `{path}`:\n-\tOnly the start of YAML tags present in the README."

A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"
A_												=     "The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Summary` but it is empty.\n-\tExpected some text in section `Dataset Summary` but it is empty (text in subsections are ignored)."


A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n"
A_												=     "The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Card for My Dataset` but it is empty.\n-\tSection `Dataset Card for My Dataset` expected the following subsections: `Table of Contents`, `Dataset Description`. Found 'None'."

A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Languages\nLanguage Text\n"

A_												=     "The following issues were found for the README at `{path}`:\n-\tSection `Dataset Description` is missing subsection: `Supported Tasks and Leaderboards`."


A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\n"

A_												=     "The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Languages` but it is empty."

A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"
A_												=     "The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README."

A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n# Dataset Card My Dataset\n"

A_												=     "The following issues were found for the README at `{path}`:\n-\tThe README has several first-level headings: `Dataset Card for My Dataset`, `Dataset Card My Dataset`. Only one heading is expected. Skipping further validation for this README."

A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

A_												=     "The following issues were found for the README at `{path}`:\n-\tNo first-level heading starting with `Dataset Card for` found in README. Skipping further validation for this README."

A_												=     ""

A_												=     "The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README.\n-\tNo YAML markers are present in the README."

A_												=     "\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

A_												=     "The following issues were found while parsing the README at `{path}`:\n-\tMultiple sections with the same heading `Dataset Card for My Dataset` have been found. Please keep only one of these sections."





@pytest.mark.parametrize(
    "readme_md, expected_dict",     [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ],     )
def    __UpperCamelCase      (    a,     a) ->Dict:
					assert ReadMe.from_string(a,     a).to_dict() == expected_dict





@pytest.mark.parametrize(
    "readme_md, expected_error",     [
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ],     )
def    __UpperCamelCase      (    a,     a) ->List[str]:
					with pytest.raises(a,     match=re.escape(expected_error.format(path="root"))):
										lowerCamelCase__       =       ReadMe.from_string(a,     a)
										readme.validate()





@pytest.mark.parametrize(
    "readme_md, expected_error",     [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ],     )
def    __UpperCamelCase      (    a,     a) ->Optional[Any]:
					with pytest.raises(a,     match=re.escape(expected_error.format(path="root"))):
										ReadMe.from_string(a,     a)





@pytest.mark.parametrize(
    "readme_md,",     [
        (README_MULTIPLE_SAME_HEADING_1),
    ],     )
def    __UpperCamelCase      (    a) ->Union[str, Any]:
					ReadMe.from_string(a,     a,     suppress_parsing_errors=a)





@pytest.mark.parametrize(
    "readme_md, expected_dict",     [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ],     )
def    __UpperCamelCase      (    a,     a) ->List[str]:
					with tempfile.TemporaryDirectory() as tmp_dir:
										lowerCamelCase__       =       Path(a) / "README.md"
										with open(a,     "w+") as readme_file:
															readme_file.write(a)
										lowerCamelCase__       =       ReadMe.from_readme(a,     a).to_dict()
										assert out["name"] == path
										assert out["text"] == ""
										assert out["is_empty_text"]
										assert out["subsections"] == expected_dict["subsections"]





@pytest.mark.parametrize(
    "readme_md, expected_error",     [
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ],     )
def    __UpperCamelCase      (    a,     a) ->Any:
					with tempfile.TemporaryDirectory() as tmp_dir:
										lowerCamelCase__       =       Path(a) / "README.md"
										with open(a,     "w+") as readme_file:
															readme_file.write(a)
										lowerCamelCase__       =       expected_error.format(path=a)
										with pytest.raises(a,     match=re.escape(a)):
															lowerCamelCase__       =       ReadMe.from_readme(a,     a)
															readme.validate()





@pytest.mark.parametrize(
    "readme_md, expected_error",     [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ],     )
def    __UpperCamelCase      (    a,     a) ->List[Any]:
					with tempfile.TemporaryDirectory() as tmp_dir:
										lowerCamelCase__       =       Path(a) / "README.md"
										with open(a,     "w+") as readme_file:
															readme_file.write(a)
										lowerCamelCase__       =       expected_error.format(path=a)
										with pytest.raises(a,     match=re.escape(a)):
															ReadMe.from_readme(a,     a)







@pytest.mark.parametrize(
    "readme_md,",     [
        (README_MULTIPLE_SAME_HEADING_1),
    ],     )
def    __UpperCamelCase      (    a) ->int:
					with tempfile.TemporaryDirectory() as tmp_dir:
										lowerCamelCase__       =       Path(a) / "README.md"
										with open(a,     "w+") as readme_file:
															readme_file.write(a)
										ReadMe.from_readme(a,     a,     suppress_parsing_errors=a)








from collections import OrderedDict
from typing import TYPE_CHECKING, Any, List, Mapping, Optional, Union

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import TensorType, logging


if TYPE_CHECKING:
						from ...onnx.config import PatchingSpec
						from ...tokenization_utils_base import PreTrainedTokenizerBase


A_												=     logging.get_logger(__name__)

A_												=     {
    "allenai/longformer-base-4096": "https://huggingface.co/allenai/longformer-base-4096/resolve/main/config.json",
    "allenai/longformer-large-4096": "https://huggingface.co/allenai/longformer-large-4096/resolve/main/config.json",
    "allenai/longformer-large-4096-finetuned-triviaqa": (
        "https://huggingface.co/allenai/longformer-large-4096-finetuned-triviaqa/resolve/main/config.json"
    ),
    "allenai/longformer-base-4096-extra.pos.embd.only": (
        "https://huggingface.co/allenai/longformer-base-4096-extra.pos.embd.only/resolve/main/config.json"
    ),
    "allenai/longformer-large-4096-extra.pos.embd.only": (
        "https://huggingface.co/allenai/longformer-large-4096-extra.pos.embd.only/resolve/main/config.json"
    ),
}

class SCREAMING_SNAKE_CASE_ (					lowercase_   ):





					"""simple docstring"""


					A__				 =    "longformer"





					def __init__( self	,  _lowerCAmelCase = 512	,  _lowerCAmelCase = 2	,  _lowerCAmelCase = 1	,  _lowerCAmelCase = 0	,  _lowerCAmelCase = 2	,  _lowerCAmelCase = 3_0522	,  _lowerCAmelCase = 768	,  _lowerCAmelCase = 12	,  _lowerCAmelCase = 12	,  _lowerCAmelCase = 3072	,  _lowerCAmelCase = "gelu"	,  _lowerCAmelCase = 0.1	,  _lowerCAmelCase = 0.1	,  _lowerCAmelCase = 512	,  _lowerCAmelCase = 2	,  _lowerCAmelCase = 0.02	,  _lowerCAmelCase = 1E-12	,  _lowerCAmelCase = False	,  **_lowerCAmelCase	,  ):
										super().__init__(pad_token_id=_lowerCAmelCase	,  **_lowerCAmelCase	)

										lowerCamelCase__       =       attention_window
										lowerCamelCase__       =       sep_token_id
										lowerCamelCase__       =       bos_token_id
										lowerCamelCase__       =       eos_token_id
										lowerCamelCase__       =       vocab_size
										lowerCamelCase__       =       hidden_size
										lowerCamelCase__       =       num_hidden_layers
										lowerCamelCase__       =       num_attention_heads
										lowerCamelCase__       =       hidden_act
										lowerCamelCase__       =       intermediate_size
										lowerCamelCase__       =       hidden_dropout_prob
										lowerCamelCase__       =       attention_probs_dropout_prob
										lowerCamelCase__       =       max_position_embeddings
										lowerCamelCase__       =       type_vocab_size
										lowerCamelCase__       =       initializer_range
										lowerCamelCase__       =       layer_norm_eps
										lowerCamelCase__       =       onnx_export

class SCREAMING_SNAKE_CASE_ (					lowercase_   ):





					"""simple docstring"""


					def __init__( self	,  _lowerCAmelCase	,  _lowerCAmelCase = "default"	,  _lowerCAmelCase = None	):
										super().__init__(_lowerCAmelCase	,  _lowerCAmelCase	,  _lowerCAmelCase	)
										lowerCamelCase__       =       True





					@property
					def 					__magic_name__ ( self	):
										if self.task == "multiple-choice":
															lowerCamelCase__       =       {0: "batch", 1: "choice", 2: "sequence"}
										else:
															lowerCamelCase__       =       {0: "batch", 1: "sequence"}
										return OrderedDict(
										    [
										        ("input_ids", dynamic_axis),
										        ("attention_mask", dynamic_axis),
										        ("global_attention_mask", dynamic_axis),
										    ]	)





					@property
					def 					__magic_name__ ( self	):
										lowerCamelCase__       =       super().outputs
										if self.task == "default":
															lowerCamelCase__       =       {0: "batch"}
										return outputs





					@property
					def 					__magic_name__ ( self	):
										return 1E-4





					@property
					def 					__magic_name__ ( self	):
										# needs to be >= 14 to support tril operator
										return max(super().default_onnx_opset	,  14	)






					def 					__magic_name__ ( self	,  _lowerCAmelCase	,  _lowerCAmelCase = -1	,  _lowerCAmelCase = -1	,  _lowerCAmelCase = False	,  _lowerCAmelCase = None	,  ):
										lowerCamelCase__       =       super().generate_dummy_inputs(
										    preprocessor=_lowerCAmelCase	,  batch_size=_lowerCAmelCase	,  seq_length=_lowerCAmelCase	,  is_pair=_lowerCAmelCase	,  framework=_lowerCAmelCase	)
										import torch

										# for some reason, replacing this code by inputs["global_attention_mask"] = torch.randint(2, inputs["input_ids"].shape, dtype=torch.int64)
										# makes the export fail randomly
										lowerCamelCase__       =       torch.zeros_like(inputs["input_ids"]	)
										# make every second token global
										lowerCamelCase__       =       1

										return inputs

