




import math







def      _a				(						SCREAMING_SNAKE_CASE_							:    float				,				SCREAMING_SNAKE_CASE_							:    float					):

					if initial_intensity < 0:
										raise ValueError("The value of intensity cannot be negative"					)
										# handling of negative values of initial intensity
					if angle < 0 or angle > 3_60:
										raise ValueError("In Malus Law, the angle is in the range 0-360 degrees"					)
										# handling of values out of allowed range
					return initial_intensity * (math.cos(math.radians(SCREAMING_SNAKE_CASE_					)					) ** 2)


if __name__ == "__main__":
					import doctest

					doctest.testmod(name="""malus_law""")










from pathlib import Path

import fire
from tqdm import tqdm







def      _a				(						SCREAMING_SNAKE_CASE_							:    Dict="ro"				,				SCREAMING_SNAKE_CASE_							:    Union[str, Any]="en"				,				SCREAMING_SNAKE_CASE_							:    Optional[Any]="wmt16"				,				SCREAMING_SNAKE_CASE_							:    List[str]=None					):
					try:
										import datasets
					except (ModuleNotFoundError, ImportError):
										raise ImportError("run pip install datasets"					)
					__lowerCAmelCase           =	F"""{src_lang}-{tgt_lang}"""
					print(F"""Converting {dataset}-{pair}"""					)
					__lowerCAmelCase           =	datasets.load_dataset(SCREAMING_SNAKE_CASE_				,				SCREAMING_SNAKE_CASE_					)
					if save_dir is None:
										__lowerCAmelCase           =	F"""{dataset}-{pair}"""
					__lowerCAmelCase           =	Path(SCREAMING_SNAKE_CASE_					)
					save_dir.mkdir(exist_ok=SCREAMING_SNAKE_CASE_					)

					for split in ds.keys():
										print(F"""Splitting {split} with {ds[split].num_rows} records"""					)

										# to save to val.source, val.target like summary datasets
										__lowerCAmelCase           =	"val" if split == "validation" else split
										__lowerCAmelCase           =	save_dir.joinpath(F"""{fn}.source"""					)
										__lowerCAmelCase           =	save_dir.joinpath(F"""{fn}.target"""					)
										__lowerCAmelCase           =	src_path.open("w+"					)
										__lowerCAmelCase           =	tgt_path.open("w+"					)

										# reader is the bottleneck so writing one record at a time doesn't slow things down
										for x in tqdm(ds[split]					):
															__lowerCAmelCase           =	x["translation"]
															src_fp.write(ex[src_lang] + "\n"					)
															tgt_fp.write(ex[tgt_lang] + "\n"					)

					print(F"""Saved {dataset} dataset to {save_dir}"""					)


if __name__ == "__main__":
					fire.Fire(download_wmt_dataset)




