



# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from ..models.auto import AutoModelForSeqaSeqLM, AutoTokenizer
from .base import PipelineTool


lowerCAmelCase__				  =  {
    '''Acehnese Arabic''': '''ace_Arab''',
    '''Acehnese Latin''': '''ace_Latn''',
    '''Mesopotamian Arabic''': '''acm_Arab''',
    '''Ta\'izzi-Adeni Arabic''': '''acq_Arab''',
    '''Tunisian Arabic''': '''aeb_Arab''',
    '''Afrikaans''': '''afr_Latn''',
    '''South Levantine Arabic''': '''ajp_Arab''',
    '''Akan''': '''aka_Latn''',
    '''Amharic''': '''amh_Ethi''',
    '''North Levantine Arabic''': '''apc_Arab''',
    '''Modern Standard Arabic''': '''arb_Arab''',
    '''Modern Standard Arabic Romanized''': '''arb_Latn''',
    '''Najdi Arabic''': '''ars_Arab''',
    '''Moroccan Arabic''': '''ary_Arab''',
    '''Egyptian Arabic''': '''arz_Arab''',
    '''Assamese''': '''asm_Beng''',
    '''Asturian''': '''ast_Latn''',
    '''Awadhi''': '''awa_Deva''',
    '''Central Aymara''': '''ayr_Latn''',
    '''South Azerbaijani''': '''azb_Arab''',
    '''North Azerbaijani''': '''azj_Latn''',
    '''Bashkir''': '''bak_Cyrl''',
    '''Bambara''': '''bam_Latn''',
    '''Balinese''': '''ban_Latn''',
    '''Belarusian''': '''bel_Cyrl''',
    '''Bemba''': '''bem_Latn''',
    '''Bengali''': '''ben_Beng''',
    '''Bhojpuri''': '''bho_Deva''',
    '''Banjar Arabic''': '''bjn_Arab''',
    '''Banjar Latin''': '''bjn_Latn''',
    '''Standard Tibetan''': '''bod_Tibt''',
    '''Bosnian''': '''bos_Latn''',
    '''Buginese''': '''bug_Latn''',
    '''Bulgarian''': '''bul_Cyrl''',
    '''Catalan''': '''cat_Latn''',
    '''Cebuano''': '''ceb_Latn''',
    '''Czech''': '''ces_Latn''',
    '''Chokwe''': '''cjk_Latn''',
    '''Central Kurdish''': '''ckb_Arab''',
    '''Crimean Tatar''': '''crh_Latn''',
    '''Welsh''': '''cym_Latn''',
    '''Danish''': '''dan_Latn''',
    '''German''': '''deu_Latn''',
    '''Southwestern Dinka''': '''dik_Latn''',
    '''Dyula''': '''dyu_Latn''',
    '''Dzongkha''': '''dzo_Tibt''',
    '''Greek''': '''ell_Grek''',
    '''English''': '''eng_Latn''',
    '''Esperanto''': '''epo_Latn''',
    '''Estonian''': '''est_Latn''',
    '''Basque''': '''eus_Latn''',
    '''Ewe''': '''ewe_Latn''',
    '''Faroese''': '''fao_Latn''',
    '''Fijian''': '''fij_Latn''',
    '''Finnish''': '''fin_Latn''',
    '''Fon''': '''fon_Latn''',
    '''French''': '''fra_Latn''',
    '''Friulian''': '''fur_Latn''',
    '''Nigerian Fulfulde''': '''fuv_Latn''',
    '''Scottish Gaelic''': '''gla_Latn''',
    '''Irish''': '''gle_Latn''',
    '''Galician''': '''glg_Latn''',
    '''Guarani''': '''grn_Latn''',
    '''Gujarati''': '''guj_Gujr''',
    '''Haitian Creole''': '''hat_Latn''',
    '''Hausa''': '''hau_Latn''',
    '''Hebrew''': '''heb_Hebr''',
    '''Hindi''': '''hin_Deva''',
    '''Chhattisgarhi''': '''hne_Deva''',
    '''Croatian''': '''hrv_Latn''',
    '''Hungarian''': '''hun_Latn''',
    '''Armenian''': '''hye_Armn''',
    '''Igbo''': '''ibo_Latn''',
    '''Ilocano''': '''ilo_Latn''',
    '''Indonesian''': '''ind_Latn''',
    '''Icelandic''': '''isl_Latn''',
    '''Italian''': '''ita_Latn''',
    '''Javanese''': '''jav_Latn''',
    '''Japanese''': '''jpn_Jpan''',
    '''Kabyle''': '''kab_Latn''',
    '''Jingpho''': '''kac_Latn''',
    '''Kamba''': '''kam_Latn''',
    '''Kannada''': '''kan_Knda''',
    '''Kashmiri Arabic''': '''kas_Arab''',
    '''Kashmiri Devanagari''': '''kas_Deva''',
    '''Georgian''': '''kat_Geor''',
    '''Central Kanuri Arabic''': '''knc_Arab''',
    '''Central Kanuri Latin''': '''knc_Latn''',
    '''Kazakh''': '''kaz_Cyrl''',
    '''Kabiyè''': '''kbp_Latn''',
    '''Kabuverdianu''': '''kea_Latn''',
    '''Khmer''': '''khm_Khmr''',
    '''Kikuyu''': '''kik_Latn''',
    '''Kinyarwanda''': '''kin_Latn''',
    '''Kyrgyz''': '''kir_Cyrl''',
    '''Kimbundu''': '''kmb_Latn''',
    '''Northern Kurdish''': '''kmr_Latn''',
    '''Kikongo''': '''kon_Latn''',
    '''Korean''': '''kor_Hang''',
    '''Lao''': '''lao_Laoo''',
    '''Ligurian''': '''lij_Latn''',
    '''Limburgish''': '''lim_Latn''',
    '''Lingala''': '''lin_Latn''',
    '''Lithuanian''': '''lit_Latn''',
    '''Lombard''': '''lmo_Latn''',
    '''Latgalian''': '''ltg_Latn''',
    '''Luxembourgish''': '''ltz_Latn''',
    '''Luba-Kasai''': '''lua_Latn''',
    '''Ganda''': '''lug_Latn''',
    '''Luo''': '''luo_Latn''',
    '''Mizo''': '''lus_Latn''',
    '''Standard Latvian''': '''lvs_Latn''',
    '''Magahi''': '''mag_Deva''',
    '''Maithili''': '''mai_Deva''',
    '''Malayalam''': '''mal_Mlym''',
    '''Marathi''': '''mar_Deva''',
    '''Minangkabau Arabic ''': '''min_Arab''',
    '''Minangkabau Latin''': '''min_Latn''',
    '''Macedonian''': '''mkd_Cyrl''',
    '''Plateau Malagasy''': '''plt_Latn''',
    '''Maltese''': '''mlt_Latn''',
    '''Meitei Bengali''': '''mni_Beng''',
    '''Halh Mongolian''': '''khk_Cyrl''',
    '''Mossi''': '''mos_Latn''',
    '''Maori''': '''mri_Latn''',
    '''Burmese''': '''mya_Mymr''',
    '''Dutch''': '''nld_Latn''',
    '''Norwegian Nynorsk''': '''nno_Latn''',
    '''Norwegian Bokmål''': '''nob_Latn''',
    '''Nepali''': '''npi_Deva''',
    '''Northern Sotho''': '''nso_Latn''',
    '''Nuer''': '''nus_Latn''',
    '''Nyanja''': '''nya_Latn''',
    '''Occitan''': '''oci_Latn''',
    '''West Central Oromo''': '''gaz_Latn''',
    '''Odia''': '''ory_Orya''',
    '''Pangasinan''': '''pag_Latn''',
    '''Eastern Panjabi''': '''pan_Guru''',
    '''Papiamento''': '''pap_Latn''',
    '''Western Persian''': '''pes_Arab''',
    '''Polish''': '''pol_Latn''',
    '''Portuguese''': '''por_Latn''',
    '''Dari''': '''prs_Arab''',
    '''Southern Pashto''': '''pbt_Arab''',
    '''Ayacucho Quechua''': '''quy_Latn''',
    '''Romanian''': '''ron_Latn''',
    '''Rundi''': '''run_Latn''',
    '''Russian''': '''rus_Cyrl''',
    '''Sango''': '''sag_Latn''',
    '''Sanskrit''': '''san_Deva''',
    '''Santali''': '''sat_Olck''',
    '''Sicilian''': '''scn_Latn''',
    '''Shan''': '''shn_Mymr''',
    '''Sinhala''': '''sin_Sinh''',
    '''Slovak''': '''slk_Latn''',
    '''Slovenian''': '''slv_Latn''',
    '''Samoan''': '''smo_Latn''',
    '''Shona''': '''sna_Latn''',
    '''Sindhi''': '''snd_Arab''',
    '''Somali''': '''som_Latn''',
    '''Southern Sotho''': '''sot_Latn''',
    '''Spanish''': '''spa_Latn''',
    '''Tosk Albanian''': '''als_Latn''',
    '''Sardinian''': '''srd_Latn''',
    '''Serbian''': '''srp_Cyrl''',
    '''Swati''': '''ssw_Latn''',
    '''Sundanese''': '''sun_Latn''',
    '''Swedish''': '''swe_Latn''',
    '''Swahili''': '''swh_Latn''',
    '''Silesian''': '''szl_Latn''',
    '''Tamil''': '''tam_Taml''',
    '''Tatar''': '''tat_Cyrl''',
    '''Telugu''': '''tel_Telu''',
    '''Tajik''': '''tgk_Cyrl''',
    '''Tagalog''': '''tgl_Latn''',
    '''Thai''': '''tha_Thai''',
    '''Tigrinya''': '''tir_Ethi''',
    '''Tamasheq Latin''': '''taq_Latn''',
    '''Tamasheq Tifinagh''': '''taq_Tfng''',
    '''Tok Pisin''': '''tpi_Latn''',
    '''Tswana''': '''tsn_Latn''',
    '''Tsonga''': '''tso_Latn''',
    '''Turkmen''': '''tuk_Latn''',
    '''Tumbuka''': '''tum_Latn''',
    '''Turkish''': '''tur_Latn''',
    '''Twi''': '''twi_Latn''',
    '''Central Atlas Tamazight''': '''tzm_Tfng''',
    '''Uyghur''': '''uig_Arab''',
    '''Ukrainian''': '''ukr_Cyrl''',
    '''Umbundu''': '''umb_Latn''',
    '''Urdu''': '''urd_Arab''',
    '''Northern Uzbek''': '''uzn_Latn''',
    '''Venetian''': '''vec_Latn''',
    '''Vietnamese''': '''vie_Latn''',
    '''Waray''': '''war_Latn''',
    '''Wolof''': '''wol_Latn''',
    '''Xhosa''': '''xho_Latn''',
    '''Eastern Yiddish''': '''ydd_Hebr''',
    '''Yoruba''': '''yor_Latn''',
    '''Yue Chinese''': '''yue_Hant''',
    '''Chinese Simplified''': '''zho_Hans''',
    '''Chinese Traditional''': '''zho_Hant''',
    '''Standard Malay''': '''zsm_Latn''',
    '''Zulu''': '''zul_Latn''',
}

class 				snake_case__(snake_case_      ):

					"""simple docstring"""






					lowercase_								=		"""facebook/nllb-200-distilled-600M"""
					lowercase_								=		(
					    """This is a tool that translates text from a language to another. It takes three inputs: `text`, which should """
					    """be the text to translate, `src_lang`, which should be the language of the text to translate and `tgt_lang`, """
					    """which should be the language for the desired ouput language. Both `src_lang` and `tgt_lang` are written in """
					    """plain English, such as \'Romanian\', or \'Albanian\'. It returns the text translated in `tgt_lang`."""
					)
					lowercase_								=		"""translator"""
					lowercase_								=		AutoTokenizer
					lowercase_								=		AutoModelForSeqaSeqLM
					lowercase_								=		LANGUAGE_CODES

					lowercase_								=		["""text""", """text""", """text"""]
					lowercase_								=		["""text"""]







					def 			snake_case					(							self :     Dict				,       SCREAMING_SNAKE_CASE :     List[Any]				,       SCREAMING_SNAKE_CASE :     List[Any]				,       SCREAMING_SNAKE_CASE :     Dict					):
									if src_lang not in self.lang_to_code:
													raise ValueError(f"""{src_lang} is not a supported language."""					)
									if tgt_lang not in self.lang_to_code:
													raise ValueError(f"""{tgt_lang} is not a supported language."""					)
									lowercase__					:   Optional[int]													=					self.lang_to_code[src_lang]
									lowercase__					:   List[str]													=					self.lang_to_code[tgt_lang]
									return self.pre_processor._build_translation_inputs(
									    SCREAMING_SNAKE_CASE				,       return_tensors="pt"				,       src_lang=SCREAMING_SNAKE_CASE				,       tgt_lang=SCREAMING_SNAKE_CASE					)







					def 			snake_case					(							self :     Optional[Any]				,       SCREAMING_SNAKE_CASE :     Union[str, Any]					):
									return self.model.generate(**SCREAMING_SNAKE_CASE					)







					def 			snake_case					(							self :     Tuple				,       SCREAMING_SNAKE_CASE :     Tuple					):
									return self.post_processor.decode(outputs[0].tolist()				,       skip_special_tokens=SCREAMING_SNAKE_CASE					)









import os
from argparse import ArgumentParser, Namespace

from ..data import SingleSentenceClassificationProcessor as Processor
from ..pipelines import TextClassificationPipeline
from ..utils import is_tf_available, is_torch_available, logging
from . import BaseTransformersCLICommand


if not is_tf_available() and not is_torch_available():
				raise RuntimeError('''At least one of PyTorch or TensorFlow 2.0+ should be installed to use CLI training''')

# TF training parameters
lowerCAmelCase__					  =  False
lowerCAmelCase__					  =  False


def  __lowerCamelCase (       lowerCamelCase__						):

				"""simple docstring"""


				return TrainCommand(lowerCamelCase__						)

class 				snake_case__(_UpperCamelCase      ):

					"""simple docstring"""






					@staticmethod
					def 			snake_case					(							SCREAMING_SNAKE_CASE :     ArgumentParser					):
									lowercase__					:   Optional[int]													=					parser.add_parser("train"				,       help="CLI tool to train a model on a task."					)

									train_parser.add_argument(
									    "--train_data"				,       type=SCREAMING_SNAKE_CASE				,       required=SCREAMING_SNAKE_CASE				,       help="path to train (and optionally evaluation) dataset as a csv with tab separated labels and sentences."				,       )
									train_parser.add_argument(
									    "--column_label"				,       type=SCREAMING_SNAKE_CASE				,       default=0				,       help="Column of the dataset csv file with example labels."					)
									train_parser.add_argument(
									    "--column_text"				,       type=SCREAMING_SNAKE_CASE				,       default=1				,       help="Column of the dataset csv file with example texts."					)
									train_parser.add_argument(
									    "--column_id"				,       type=SCREAMING_SNAKE_CASE				,       default=2				,       help="Column of the dataset csv file with example ids."					)
									train_parser.add_argument(
									    "--skip_first_row"				,       action="store_true"				,       help="Skip the first row of the csv file (headers)."					)

									train_parser.add_argument("--validation_data"				,       type=SCREAMING_SNAKE_CASE				,       default=""				,       help="path to validation dataset."					)
									train_parser.add_argument(
									    "--validation_split"				,       type=SCREAMING_SNAKE_CASE				,       default=0.1				,       help="if validation dataset is not provided, fraction of train dataset to use as validation dataset."				,       )

									train_parser.add_argument("--output"				,       type=SCREAMING_SNAKE_CASE				,       default="./"				,       help="path to saved the trained model."					)

									train_parser.add_argument(
									    "--task"				,       type=SCREAMING_SNAKE_CASE				,       default="text_classification"				,       help="Task to train the model on."					)
									train_parser.add_argument(
									    "--model"				,       type=SCREAMING_SNAKE_CASE				,       default="bert-base-uncased"				,       help="Model's name or path to stored model."					)
									train_parser.add_argument("--train_batch_size"				,       type=SCREAMING_SNAKE_CASE				,       default=32				,       help="Batch size for training."					)
									train_parser.add_argument("--valid_batch_size"				,       type=SCREAMING_SNAKE_CASE				,       default=64				,       help="Batch size for validation."					)
									train_parser.add_argument("--learning_rate"				,       type=SCREAMING_SNAKE_CASE				,       default=3E-5				,       help="Learning rate."					)
									train_parser.add_argument("--adam_epsilon"				,       type=SCREAMING_SNAKE_CASE				,       default=1E-0_8				,       help="Epsilon for Adam optimizer."					)
									train_parser.set_defaults(func=SCREAMING_SNAKE_CASE					)







					def __init__(							self :     int				,       SCREAMING_SNAKE_CASE :     Namespace					):
									lowercase__					:   int													=					logging.get_logger("transformers-cli/training"					)

									lowercase__					:   List[Any]													=					"tf" if is_tf_available() else "torch"

									os.makedirs(args.output				,       exist_ok=SCREAMING_SNAKE_CASE					)
									lowercase__					:   Optional[Any]													=					args.output

									lowercase__					:   Union[str, Any]													=					args.column_label
									lowercase__					:   Optional[int]													=					args.column_text
									lowercase__					:   Optional[int]													=					args.column_id

									self.logger.info(f"""Loading {args.task} pipeline for {args.model}"""					)
									if args.task == "text_classification":
													lowercase__					:   int													=					TextClassificationPipeline.from_pretrained(args.model					)
									elif args.task == "token_classification":
													raise NotImplementedError
									elif args.task == "question_answering":
													raise NotImplementedError

									self.logger.info(f"""Loading dataset from {args.train_data}"""					)
									lowercase__					:   List[str]													=					Processor.create_from_csv(
									    args.train_data				,       column_label=args.column_label				,       column_text=args.column_text				,       column_id=args.column_id				,       skip_first_row=args.skip_first_row				,       )
									lowercase__					:   Union[str, Any]													=					None
									if args.validation_data:
													self.logger.info(f"""Loading validation dataset from {args.validation_data}"""					)
													lowercase__					:   Optional[int]													=					Processor.create_from_csv(
													    args.validation_data				,       column_label=args.column_label				,       column_text=args.column_text				,       column_id=args.column_id				,       skip_first_row=args.skip_first_row				,       )

									lowercase__					:   Dict													=					args.validation_split
									lowercase__					:   List[str]													=					args.train_batch_size
									lowercase__					:   Any													=					args.valid_batch_size
									lowercase__					:   Optional[int]													=					args.learning_rate
									lowercase__					:   int													=					args.adam_epsilon







					def 			snake_case					(							self :     Dict					):
									if self.framework == "tf":
													return self.run_tf()
									return self.run_torch()







					def 			snake_case					(							self :     Union[str, Any]					):
									raise NotImplementedError







					def 			snake_case					(							self :     Union[str, Any]					):
									self.pipeline.fit(
									    self.train_dataset				,       validation_data=self.valid_dataset				,       validation_split=self.validation_split				,       learning_rate=self.learning_rate				,       adam_epsilon=self.adam_epsilon				,       train_batch_size=self.train_batch_size				,       valid_batch_size=self.valid_batch_size				,       )

									# Save trained pipeline
									self.pipeline.save_pretrained(self.output					)




