






import subprocess
from argparse import ArgumentParser
from typing import List, Union

from huggingface_hub.hf_api import HfFolder, create_repo, whoami
from requests.exceptions import HTTPError

from . import BaseTransformersCLICommand






class 	lowercase_      (			lowercase     ):






      '''simple docstring'''




      @staticmethod
      def    __lowerCAmelCase      (   __UpperCAmelCase   :		ArgumentParser						)					->Dict:




       """simple docstring"""


       a							=   parser.add_parser('''login'''				,      help='''Log in using the same credentials as on huggingface.co'''						)
       login_parser.set_defaults(func=lambda __UpperCAmelCase						: LoginCommand(__UpperCAmelCase						)						)
       a							=   parser.add_parser('''whoami'''				,      help='''Find out which huggingface.co account you are logged in as.'''						)
       whoami_parser.set_defaults(func=lambda __UpperCAmelCase						: WhoamiCommand(__UpperCAmelCase						)						)
       a							=   parser.add_parser('''logout'''				,      help='''Log out'''						)
       logout_parser.set_defaults(func=lambda __UpperCAmelCase						: LogoutCommand(__UpperCAmelCase						)						)

       # new system: git-based repo system
       a							=   parser.add_parser(
           '''repo'''				,      help='''Deprecated: use `huggingface-cli` instead. Commands to interact with your huggingface.co repos.'''				,      )
       a							=   repo_parser.add_subparsers(
           help='''Deprecated: use `huggingface-cli` instead. huggingface.co repos related commands'''						)
       a							=   repo_subparsers.add_parser(
           '''create'''				,      help='''Deprecated: use `huggingface-cli` instead. Create a new repo on huggingface.co'''						)
       repo_create_parser.add_argument(
           '''name'''				,      type=__UpperCAmelCase				,      help='''Name for your model\'s repo. Will be namespaced under your username to build the model id.'''				,      )
       repo_create_parser.add_argument('''--organization'''				,      type=__UpperCAmelCase				,      help='''Optional: organization namespace.'''						)
       repo_create_parser.add_argument('''-y'''				,      '''--yes'''				,      action='''store_true'''				,      help='''Optional: answer Yes to the prompt'''						)
       repo_create_parser.set_defaults(func=lambda __UpperCAmelCase						: RepoCreateCommand(__UpperCAmelCase						)						)






class 	lowercase_      :






      '''simple docstring'''




      __snake_case  						=						'''\u001b[1m'''
      __snake_case  						=						'''\u001b[31m'''
      __snake_case  						=						'''\u001b[90m'''
      __snake_case  						=						'''\u001b[0m'''







      @classmethod
      def    __lowerCAmelCase      (   cls   :		Tuple				,      __UpperCAmelCase   :		str						)					->Optional[int]:




       """simple docstring"""


       return F"""{cls._bold}{s}{cls._reset}"""







      @classmethod
      def    __lowerCAmelCase      (   cls   :		List[Any]				,      __UpperCAmelCase   :		Optional[Any]						)					->Any:




       """simple docstring"""


       return F"""{cls._bold}{cls._red}{s}{cls._reset}"""







      @classmethod
      def    __lowerCAmelCase      (   cls   :		Union[str, Any]				,      __UpperCAmelCase   :		Tuple						)					->int:




       """simple docstring"""


       return F"""{cls._gray}{s}{cls._reset}"""
def 		_a					(   a				:List[List[Union[str, int]]]			,							a				:List[str]				)							->    str:
 a							=   [max(len(str(a				)				) for x in col				) for col in zip(*a			,							a				)]
 a							=   ('''{{:{}}} ''' * len(a				)).format(*a				)
 a							=   []
 lines.append(row_format.format(*a				)				)
 lines.append(row_format.format(*['''-''' * w for w in col_widths]				)				)
 for row in rows:
  lines.append(row_format.format(*a				)				)
 return "\n".join(a				)






class 	lowercase_      :






      '''simple docstring'''




      def __init__(   self   :		Optional[int]				,      __UpperCAmelCase   :		Union[str, Any]						)					->Union[str, Any]:




       """simple docstring"""


       a							=   args






class 	lowercase_      (			lowercase     ):






      '''simple docstring'''




      def    __lowerCAmelCase      (   self   :		List[str]						)					->str:




       """simple docstring"""


       print(
           ANSI.red(
               '''ERROR! `huggingface-cli login` uses an outdated login mechanism '''
               '''that is not compatible with the Hugging Face Hub backend anymore. '''
               '''Please use `huggingface-cli login instead.'''						)						)






class 	lowercase_      (			lowercase     ):






      '''simple docstring'''




      def    __lowerCAmelCase      (   self   :		List[Any]						)					->List[Any]:




       """simple docstring"""


       print(
           ANSI.red(
               '''WARNING! `transformers-cli whoami` is deprecated and will be removed in v5. Please use '''
               '''`huggingface-cli whoami` instead.'''						)						)
       a							=   HfFolder.get_token()
       if token is None:
        print('''Not logged in'''						)
        exit()
       try:
        a						,  a							=   whoami(__UpperCAmelCase						)
        print(__UpperCAmelCase						)
        if orgs:
         print(ANSI.bold('''orgs: '''						)				,      ''','''.join(__UpperCAmelCase						)						)
       except HTTPError as e:
        print(__UpperCAmelCase						)
        print(ANSI.red(e.response.text						)						)
        exit(1						)






class 	lowercase_      (			lowercase     ):






      '''simple docstring'''




      def    __lowerCAmelCase      (   self   :		Dict						)					->List[Any]:




       """simple docstring"""


       print(
           ANSI.red(
               '''ERROR! `transformers-cli logout` uses an outdated logout mechanism '''
               '''that is not compatible with the Hugging Face Hub backend anymore. '''
               '''Please use `huggingface-cli logout instead.'''						)						)






class 	lowercase_      (			lowercase     ):






      '''simple docstring'''




      def    __lowerCAmelCase      (   self   :		List[str]						)					->Union[str, Any]:




       """simple docstring"""


       print(
           ANSI.red(
               '''WARNING! Managing repositories through transformers-cli is deprecated. '''
               '''Please use `huggingface-cli` instead.'''						)						)
       a							=   HfFolder.get_token()
       if token is None:
        print('''Not logged in'''						)
        exit(1						)
       try:
        a							=   subprocess.check_output(['''git''', '''--version''']						).decode('''utf-8'''						)
        print(ANSI.gray(stdout.strip()						)						)
       except FileNotFoundError:
        print('''Looks like you do not have git installed, please install.'''						)

       try:
        a							=   subprocess.check_output(['''git-lfs''', '''--version''']						).decode('''utf-8'''						)
        print(ANSI.gray(stdout.strip()						)						)
       except FileNotFoundError:
        print(
            ANSI.red(
                '''Looks like you do not have git-lfs installed, please install.'''
                ''' You can install from https://git-lfs.github.com/.'''
                ''' Then run `git lfs install` (you only have to do this once).'''						)						)
       print(''''''						)

       a						,  a							=   whoami(__UpperCAmelCase						)
       a							=   self.args.organization if self.args.organization is not None else user
       a							=   F"""{namespace}/{self.args.name}"""
       print(F"""You are about to create {ANSI.bold(__UpperCAmelCase						)}"""						)

       if not self.args.yes:
        a							=   input('''Proceed? [Y/n] '''						).lower()
        if not (choice == "" or choice == "y" or choice == "yes"):
         print('''Abort'''						)
         exit()
       try:
        a							=   create_repo(__UpperCAmelCase				,      name=self.args.name				,      organization=self.args.organization						)
       except HTTPError as e:
        print(__UpperCAmelCase						)
        print(ANSI.red(e.response.text						)						)
        exit(1						)
       print('''\nYour repo now lives at:'''						)
       print(F"""  {ANSI.bold(__UpperCAmelCase						)}"""						)
       print('''\nYou can clone it locally with the command below, and commit/push as usual.'''						)
       print(F"""\n  git clone {url}"""						)
       print(''''''						)










import enum
import inspect
from typing import Iterable, List, Optional, Tuple, Union






class 	lowercase_      (			enum.Enum     ):






      '''simple docstring'''




      __snake_case  						=						'''timm'''
      __snake_case  						=						'''transformers'''
def 		_a					(   a				:Optional[Iterable[str]]			,							a				:Optional[Iterable[int]]			,							a				:Optional[Iterable[str]]				)							->    Optional[Any]:
 if stage_names is None:
  raise ValueError('''Stage_names must be set for transformers backbones'''				)

 if out_features is not None:
  if not isinstance(a			,							(list,)				):
   raise ValueError(F"""out_features must be a list {type(a				)}"""				)
  if any(feat not in stage_names for feat in out_features				):
   raise ValueError(F"""out_features must be a subset of stage_names: {stage_names} got {out_features}"""				)

 if out_indices is not None:
  if not isinstance(a			,							(list, tuple)				):
   raise ValueError(F"""out_indices must be a list or tuple, got {type(a				)}"""				)
  if any(idx >= len(a				) for idx in out_indices				):
   raise ValueError('''out_indices must be valid indices for stage_names {stage_names}, got {out_indices}'''				)

 if out_features is not None and out_indices is not None:
  if len(a				) != len(a				):
   raise ValueError('''out_features and out_indices should have the same length if both are set'''				)
  if out_features != [stage_names[idx] for idx in out_indices]:
   raise ValueError('''out_features and out_indices should correspond to the same stages if both are set'''				)
def 		_a					(   a				:Optional[List[str]]			,							a				:Optional[Union[List[int], Tuple[int]]]			,							a				:List[str]			,							)							->    Union[str, Any]:
 if out_indices is None and out_features is None:
  a							=   [len(a				) - 1]
  a							=   [stage_names[-1]]
 elif out_indices is None and out_features is not None:
  a							=   [stage_names.index(a				) for layer in out_features]
 elif out_features is None and out_indices is not None:
  a							=   [stage_names[idx] for idx in out_indices]
 return out_features, out_indices
def 		_a					(   a				:Optional[List[str]]			,							a				:Optional[Union[List[int], Tuple[int]]]			,							a				:List[str]			,							)							->    Tuple[List[str], List[int]]:
 verify_out_features_out_indices(out_features=a			,							out_indices=a			,							stage_names=a				)
 a						,  a							=   _align_output_features_output_indices(
     out_features=a			,							out_indices=a			,							stage_names=a				)
 # Verify that the aligned out_features and out_indices are valid
 verify_out_features_out_indices(out_features=a			,							out_indices=a			,							stage_names=a				)
 return output_features, output_indices






class 	lowercase_      :






      '''simple docstring'''




      __snake_case  						=						None







      def    __lowerCAmelCase      (   self   :		Any				,      __UpperCAmelCase   :		Tuple						)					->None:




       """simple docstring"""


       if getattr(self				,      '''_backbone'''				,      __UpperCAmelCase						) is None:
        raise ValueError('''self._backbone must be set before calling _init_timm_backbone'''						)

       # These will diagree with the defaults for the transformers models e.g. for resnet50
       # the transformer model has out_features = ['stem', 'stage1', 'stage2', 'stage3', 'stage4']
       # the timm model has out_features = ['act', 'layer1', 'layer2', 'layer3', 'layer4']
       a							=   [stage['''module'''] for stage in self._backbone.feature_info.info]
       a							=   [stage['''num_chs'''] for stage in self._backbone.feature_info.info]
       a							=   self._backbone.feature_info.out_indices
       a							=   self._backbone.feature_info.module_name()

       # We verify the out indices and out features are valid
       verify_out_features_out_indices(
           out_features=__UpperCAmelCase				,      out_indices=__UpperCAmelCase				,      stage_names=self.stage_names						)
       a						,  a							=   out_features, out_indices







      def    __lowerCAmelCase      (   self   :		Any				,      __UpperCAmelCase   :		Tuple						)					->None:




       """simple docstring"""


       a							=   getattr(__UpperCAmelCase				,      '''stage_names'''						)
       a							=   getattr(__UpperCAmelCase				,      '''out_features'''				,      __UpperCAmelCase						)
       a							=   getattr(__UpperCAmelCase				,      '''out_indices'''				,      __UpperCAmelCase						)

       a							=   stage_names
       a						,  a							=   get_aligned_output_features_output_indices(
           out_features=__UpperCAmelCase				,      out_indices=__UpperCAmelCase				,      stage_names=__UpperCAmelCase						)
       # Number of channels for each stage. This is set in the transformer backbone model init
       a							=   None







      def    __lowerCAmelCase      (   self   :		str				,      __UpperCAmelCase   :		Union[str, Any]						)					->None:




       """simple docstring"""


       a							=   config

       a							=   getattr(__UpperCAmelCase				,      '''use_timm_backbone'''				,      __UpperCAmelCase						)
       a							=   BackboneType.TIMM if self.use_timm_backbone else BackboneType.TRANSFORMERS

       if self.backbone_type == BackboneType.TIMM:
        self._init_timm_backbone(__UpperCAmelCase						)
       elif self.backbone_type == BackboneType.TRANSFORMERS:
        self._init_transformers_backbone(__UpperCAmelCase						)
       else:
        raise ValueError(F"""backbone_type {self.backbone_type} not supported."""						)







      @property
      def    __lowerCAmelCase      (   self   :		Dict						)					->Optional[Any]:




       """simple docstring"""


       return self._out_features







      @out_features.setter
      def    __lowerCAmelCase      (   self   :		Tuple				,      __UpperCAmelCase   :		List[str]						)					->List[Any]:




       """simple docstring"""


       a						,  a							=   get_aligned_output_features_output_indices(
           out_features=__UpperCAmelCase				,      out_indices=__UpperCAmelCase				,      stage_names=self.stage_names						)







      @property
      def    __lowerCAmelCase      (   self   :		Dict						)					->int:




       """simple docstring"""


       return self._out_indices







      @out_indices.setter
      def    __lowerCAmelCase      (   self   :		Tuple				,      __UpperCAmelCase   :		Union[Tuple[int], List[int]]						)					->List[str]:




       """simple docstring"""


       a						,  a							=   get_aligned_output_features_output_indices(
           out_features=__UpperCAmelCase				,      out_indices=__UpperCAmelCase				,      stage_names=self.stage_names						)







      @property
      def    __lowerCAmelCase      (   self   :		Tuple						)					->str:




       """simple docstring"""


       return {stage: self.num_features[i] for i, stage in enumerate(self.stage_names						)}







      @property
      def    __lowerCAmelCase      (   self   :		int						)					->Optional[int]:




       """simple docstring"""


       return [self.out_feature_channels[name] for name in self.out_features]







      def    __lowerCAmelCase      (   self   :		Optional[Any]				,      *__UpperCAmelCase   :		List[str]				,      **__UpperCAmelCase   :		Optional[Any]						)					->Optional[int]:




       """simple docstring"""


       a							=   dict(inspect.signature(self.forward						).parameters						)
       a							=   {k: v for k, v in kwargs.items() if k in signature}
       return self(*__UpperCAmelCase				,      **__UpperCAmelCase						)







      def    __lowerCAmelCase      (   self   :		List[str]				,      __UpperCAmelCase   :		Dict				,      __UpperCAmelCase   :		Optional[bool] = None				,      __UpperCAmelCase   :		Optional[bool] = None				,      __UpperCAmelCase   :		Optional[bool] = None				,      )					->int:




       """simple docstring"""


       raise NotImplementedError('''This method should be implemented by the derived class.'''						)







      def    __lowerCAmelCase      (   self   :		Union[str, Any]						)					->Optional[Any]:




       """simple docstring"""


       a							=   super().to_dict()
       a							=   output.pop('''_out_features'''						)
       a							=   output.pop('''_out_indices'''						)
       return output






class 	lowercase_      :






      '''simple docstring'''




      @property
      def    __lowerCAmelCase      (   self   :		List[str]						)					->Optional[int]:




       """simple docstring"""


       return self._out_features







      @out_features.setter
      def    __lowerCAmelCase      (   self   :		int				,      __UpperCAmelCase   :		List[str]						)					->int:




       """simple docstring"""


       a						,  a							=   get_aligned_output_features_output_indices(
           out_features=__UpperCAmelCase				,      out_indices=__UpperCAmelCase				,      stage_names=self.stage_names						)







      @property
      def    __lowerCAmelCase      (   self   :		str						)					->str:




       """simple docstring"""


       return self._out_indices







      @out_indices.setter
      def    __lowerCAmelCase      (   self   :		str				,      __UpperCAmelCase   :		Union[Tuple[int], List[int]]						)					->int:




       """simple docstring"""


       a						,  a							=   get_aligned_output_features_output_indices(
           out_features=__UpperCAmelCase				,      out_indices=__UpperCAmelCase				,      stage_names=self.stage_names						)







      def    __lowerCAmelCase      (   self   :		Dict						)					->List[Any]:




       """simple docstring"""


       a							=   super().to_dict()
       a							=   output.pop('''_out_features'''						)
       a							=   output.pop('''_out_indices'''						)
       return output


