


"""simple docstring"""




import pytest

from datasets.splits import SplitDict, SplitInfo
from datasets.utils.py_utils import asdict

@pytest.mark.parametrize(
    '''split_dict'''    ,					[
        SplitDict(),
        SplitDict({'''train''': SplitInfo(name='''train'''    ,					num_bytes=1337    ,					num_examples=42    ,					dataset_name='''my_dataset'''			)}			),
        SplitDict({'''train''': SplitInfo(name='''train'''    ,					num_bytes=1337    ,					num_examples=42			)}			),
        SplitDict({'''train''': SplitInfo()}			),
    ]    ,					)
def        lowercase							(		lowerCAmelCase__ :     SplitDict			) ->      List[Any]:
		__a           =			split_dict._to_yaml_list()
		assert len(lowerCAmelCase__			) == len(lowerCAmelCase__			)
		__a           =			SplitDict._from_yaml_list(lowerCAmelCase__			)
		for split_name, split_info in split_dict.items():
				# dataset_name field is deprecated, and is therefore not part of the YAML dump
				__a           =			None
				# the split name of split_dict takes over the name of the split info object
				__a           =			split_name
		assert split_dict == reloaded




@pytest.mark.parametrize(
    '''split_info'''    ,					[SplitInfo(), SplitInfo(dataset_name=lowerCAmelCase__			), SplitInfo(dataset_name='''my_dataset'''			)]			)
def        lowercase							(		lowerCAmelCase__ :     int			) ->      List[str]:
		# For backward compatibility, we need asdict(split_dict) to return split info dictrionaries with the "dataset_name"
		# field even if it's deprecated. This way old versionso of `datasets` can still reload dataset_infos.json files
		__a           =			asdict(SplitDict({'''train''': split_info}			)			)
		assert "dataset_name" in split_dict_asdict["train"]
		assert split_dict_asdict["train"]["dataset_name"] == split_info.dataset_name










"""simple docstring"""




import numpy as np

def        lowercase							(		lowerCAmelCase__ :     np.ndarray    ,					lowerCAmelCase__ :     float			) ->      np.ndarray:
		return np.where(vector > 0    ,					lowerCAmelCase__    ,					(alpha * (np.exp(lowerCAmelCase__			) - 1))			)


if __name__ == "__main__":
				import doctest

				doctest.testmod()






