






"""simple docstring"""







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       :	Union[str, Any]             = 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       :	List[str]             = {
    '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       :	Optional[int]             = '\\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       :	Dict             = '\\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       :	Union[str, Any]             = {
    '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       :	Optional[int]             = '\\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       :	Optional[int]             = (
    'The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README.'
)

__a       :	Any             = '\\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       :	Tuple             = (
    'The following issues were found for the README at `{path}`:\n-\tNo YAML markers are present in the README.'
)

__a       :	str             = '\\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       :	int             = 'The following issues were found for the README at `{path}`:\n-\tOnly the start of YAML tags present in the README.'

__a       :	Optional[int]             = '\\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       :	Tuple             = '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       :	Any             = '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n'
__a       :	Dict             = '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       :	Optional[Any]             = '\\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       :	Optional[int]             = 'The following issues were found for the README at `{path}`:\n-\tSection `Dataset Description` is missing subsection: `Supported Tasks and Leaderboards`.'


__a       :	Optional[int]             = '\\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       :	Optional[Any]             = 'The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Languages` but it is empty.'

__a       :	str             = '\\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       :	List[Any]             = '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       :	List[Any]             = '\\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       :	Optional[int]             = '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       :	List[str]             = '\\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       :	int             = '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       :	Tuple             = ''

__a       :	Dict             = '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       :	Any             = '\\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       :	Any             = '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        SCREAMING_SNAKE_CASE       (      lowerCamelCase_ ,     lowerCamelCase_):
					assert ReadMe.from_string(lowerCamelCase_ ,     lowerCamelCase_).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        SCREAMING_SNAKE_CASE       (      lowerCamelCase_ ,     lowerCamelCase_):
					with pytest.raises(lowerCamelCase_ ,     match=re.escape(expected_error.format(path='''root'''))):
										a__          =  ReadMe.from_string(lowerCamelCase_ ,     lowerCamelCase_)
										readme.validate()





@pytest.mark.parametrize(
    '''readme_md, expected_error''' ,     [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ] ,     )
def        SCREAMING_SNAKE_CASE       (      lowerCamelCase_ ,     lowerCamelCase_):
					with pytest.raises(lowerCamelCase_ ,     match=re.escape(expected_error.format(path='''root'''))):
										ReadMe.from_string(lowerCamelCase_ ,     lowerCamelCase_)





@pytest.mark.parametrize(
    '''readme_md,''' ,     [
        (README_MULTIPLE_SAME_HEADING_1),
    ] ,     )
def        SCREAMING_SNAKE_CASE       (      lowerCamelCase_):
					ReadMe.from_string(lowerCamelCase_ ,     lowerCamelCase_ ,     suppress_parsing_errors=lowerCamelCase_)





@pytest.mark.parametrize(
    '''readme_md, expected_dict''' ,     [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ] ,     )
def        SCREAMING_SNAKE_CASE       (      lowerCamelCase_ ,     lowerCamelCase_):
					with tempfile.TemporaryDirectory() as tmp_dir:
										a__          =  Path(lowerCamelCase_) / '''README.md'''
										with open(lowerCamelCase_ ,     '''w+''') as readme_file:
															readme_file.write(lowerCamelCase_)
										a__          =  ReadMe.from_readme(lowerCamelCase_ ,     lowerCamelCase_).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        SCREAMING_SNAKE_CASE       (      lowerCamelCase_ ,     lowerCamelCase_):
					with tempfile.TemporaryDirectory() as tmp_dir:
										a__          =  Path(lowerCamelCase_) / '''README.md'''
										with open(lowerCamelCase_ ,     '''w+''') as readme_file:
															readme_file.write(lowerCamelCase_)
										a__          =  expected_error.format(path=lowerCamelCase_)
										with pytest.raises(lowerCamelCase_ ,     match=re.escape(lowerCamelCase_)):
															a__          =  ReadMe.from_readme(lowerCamelCase_ ,     lowerCamelCase_)
															readme.validate()





@pytest.mark.parametrize(
    '''readme_md, expected_error''' ,     [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ] ,     )
def        SCREAMING_SNAKE_CASE       (      lowerCamelCase_ ,     lowerCamelCase_):
					with tempfile.TemporaryDirectory() as tmp_dir:
										a__          =  Path(lowerCamelCase_) / '''README.md'''
										with open(lowerCamelCase_ ,     '''w+''') as readme_file:
															readme_file.write(lowerCamelCase_)
										a__          =  expected_error.format(path=lowerCamelCase_)
										with pytest.raises(lowerCamelCase_ ,     match=re.escape(lowerCamelCase_)):
															ReadMe.from_readme(lowerCamelCase_ ,     lowerCamelCase_)





@pytest.mark.parametrize(
    '''readme_md,''' ,     [
        (README_MULTIPLE_SAME_HEADING_1),
    ] ,     )
def        SCREAMING_SNAKE_CASE       (      lowerCamelCase_):
					with tempfile.TemporaryDirectory() as tmp_dir:
										a__          =  Path(lowerCamelCase_) / '''README.md'''
										with open(lowerCamelCase_ ,     '''w+''') as readme_file:
															readme_file.write(lowerCamelCase_)
										ReadMe.from_readme(lowerCamelCase_ ,     lowerCamelCase_ ,     suppress_parsing_errors=lowerCamelCase_)










"""simple docstring"""







from ...configuration_utils import PretrainedConfig
from ...utils import logging


__a       :	List[str]             = logging.get_logger(__name__)

__a       :	int             = {
    'facebook/vit-mae-base': 'https://huggingface.co/facebook/vit-mae-base/resolve/main/config.json',
    # See all ViT MAE models at https://huggingface.co/models?filter=vit-mae
}







class        _SCREAMING_SNAKE_CASE						(  __snake_case							):





							"""simple docstring"""





							_SCREAMING_SNAKE_CASE    ='vit_mae'




							def __init__(   self:     str				,		__A:     str=768				,		__A:     Union[str, Any]=12				,		__A:     str=12				,		__A:     List[Any]=3072				,		__A:     List[Any]="gelu"				,		__A:     Union[str, Any]=0.0				,		__A:     Tuple=0.0				,		__A:     str=0.0_2				,		__A:     Tuple=1e-12				,		__A:     Dict=224				,		__A:     List[Any]=16				,		__A:     Tuple=3				,		__A:     str=True				,		__A:     Optional[Any]=16				,		__A:     List[Any]=512				,		__A:     Optional[Any]=8				,		__A:     Dict=2048				,		__A:     str=0.7_5				,		__A:     str=False				,		**__A:     Tuple				,		):

												'''simple docstring'''

												super().__init__(**__A    )

												a__          =  hidden_size
												a__          =  num_hidden_layers
												a__          =  num_attention_heads
												a__          =  intermediate_size
												a__          =  hidden_act
												a__          =  hidden_dropout_prob
												a__          =  attention_probs_dropout_prob
												a__          =  initializer_range
												a__          =  layer_norm_eps
												a__          =  image_size
												a__          =  patch_size
												a__          =  num_channels
												a__          =  qkv_bias
												a__          =  decoder_num_attention_heads
												a__          =  decoder_hidden_size
												a__          =  decoder_num_hidden_layers
												a__          =  decoder_intermediate_size
												a__          =  mask_ratio
												a__          =  norm_pix_loss


