


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


A      = logging.get_logger(__name__)


class     lowercase__  (					__SCREAMING_SNAKE_CASE		):
  A__=  'timm_backbone'






  def __init__(					self      :					Dict		,						_lowercase      :					str=None		,						_lowercase      :					Any=3		,						_lowercase      :					Dict=True		,						_lowercase      :					Tuple=True		,						_lowercase      :					List[Any]=None		,						**_lowercase      :					Tuple		,						):




         """simple docstring"""







         super().__init__(**_lowercase )
         UpperCAmelCase__										=     backbone
         UpperCAmelCase__										=     num_channels
         UpperCAmelCase__										=     features_only
         UpperCAmelCase__										=     use_pretrained_backbone
         UpperCAmelCase__										=     True
         UpperCAmelCase__										=     out_indices if out_indices is not None else (-1,)









class     lowercase__  :






  def __init__(					self      :					List[str]		,						_lowercase      :					list ):




         """simple docstring"""







         UpperCAmelCase__										=     set_counts
         UpperCAmelCase__										=     max(_lowercase )
         UpperCAmelCase__										=     len(_lowercase )
         UpperCAmelCase__										=     [1] * num_sets
         UpperCAmelCase__										=     list(range(_lowercase ) )






  def 			_UpperCAmelCase  (					self      :					Dict		,						_lowercase      :					int		,						_lowercase      :					int ):




         """simple docstring"""







         UpperCAmelCase__										=     self.get_parent(_lowercase )
         UpperCAmelCase__										=     self.get_parent(_lowercase )

         if src_parent == dst_parent:
                return False

         if self.ranks[dst_parent] >= self.ranks[src_parent]:
                self.set_counts[dst_parent] += self.set_counts[src_parent]
                UpperCAmelCase__										=     0
                UpperCAmelCase__										=     dst_parent
                if self.ranks[dst_parent] == self.ranks[src_parent]:
                       self.ranks[dst_parent] += 1
                UpperCAmelCase__										=     self.set_counts[dst_parent]
         else:
                self.set_counts[src_parent] += self.set_counts[dst_parent]
                UpperCAmelCase__										=     0
                UpperCAmelCase__										=     src_parent
                UpperCAmelCase__										=     self.set_counts[src_parent]

         UpperCAmelCase__										=     max(self.max_set		,						_lowercase )
         return True






  def 			_UpperCAmelCase  (					self      :					Optional[Any]		,						_lowercase      :					int ):




         """simple docstring"""







         if self.parents[disj_set] == disj_set:
                return disj_set
         UpperCAmelCase__										=     self.get_parent(self.parents[disj_set] )
         return self.parents[disj_set]





