





'''simple docstring'''



import os
from itertools import chain
from random import randrange, shuffle

import pytest

from .sola import PokerHand

_SCREAMING_SNAKE_CASE					=    (
    '''4S 3H 2C 7S 5H''',
    '''9D 8H 2C 6S 7H''',
    '''2D 6D 9D TH 7D''',
    '''TC 8C 2S JH 6C''',
    '''JH 8S TH AH QH''',
    '''TS KS 5S 9S AC''',
    '''KD 6S 9D TH AD''',
    '''KS 8D 4D 9S 4S''',  # pair
    '''8C 4S KH JS 4D''',  # pair
    '''QH 8H KD JH 8S''',  # pair
    '''KC 4H KS 2H 8D''',  # pair
    '''KD 4S KC 3H 8S''',  # pair
    '''AH 8S AS KC JH''',  # pair
    '''3H 4C 4H 3S 2H''',  # 2 pairs
    '''5S 5D 2C KH KH''',  # 2 pairs
    '''3C KH 5D 5S KH''',  # 2 pairs
    '''AS 3C KH AD KH''',  # 2 pairs
    '''7C 7S 3S 7H 5S''',  # 3 of a kind
    '''7C 7S KH 2H 7H''',  # 3 of a kind
    '''AC KH QH AH AS''',  # 3 of a kind
    '''2H 4D 3C AS 5S''',  # straight (low ace)
    '''3C 5C 4C 2C 6H''',  # straight
    '''6S 8S 7S 5H 9H''',  # straight
    '''JS QS 9H TS KH''',  # straight
    '''QC KH TS JS AH''',  # straight (high ace)
    '''8C 9C 5C 3C TC''',  # flush
    '''3S 8S 9S 5S KS''',  # flush
    '''4C 5C 9C 8C KC''',  # flush
    '''JH 8H AH KH QH''',  # flush
    '''3D 2H 3H 2C 2D''',  # full house
    '''2H 2C 3S 3H 3D''',  # full house
    '''KH KC 3S 3H 3D''',  # full house
    '''JC 6H JS JD JH''',  # 4 of a kind
    '''JC 7H JS JD JH''',  # 4 of a kind
    '''JC KH JS JD JH''',  # 4 of a kind
    '''2S AS 4S 5S 3S''',  # straight flush (low ace)
    '''2D 6D 3D 4D 5D''',  # straight flush
    '''5C 6C 3C 7C 4C''',  # straight flush
    '''JH 9H TH KH QH''',  # straight flush
    '''JH AH TH KH QH''',  # royal flush (high ace straight flush)
)

_SCREAMING_SNAKE_CASE					=    (
    ('''2H 3H 4H 5H 6H''', '''KS AS TS QS JS''', '''Loss'''),
    ('''2H 3H 4H 5H 6H''', '''AS AD AC AH JD''', '''Win'''),
    ('''AS AH 2H AD AC''', '''JS JD JC JH 3D''', '''Win'''),
    ('''2S AH 2H AS AC''', '''JS JD JC JH AD''', '''Loss'''),
    ('''2S AH 2H AS AC''', '''2H 3H 5H 6H 7H''', '''Win'''),
    ('''AS 3S 4S 8S 2S''', '''2H 3H 5H 6H 7H''', '''Win'''),
    ('''2H 3H 5H 6H 7H''', '''2S 3H 4H 5S 6C''', '''Win'''),
    ('''2S 3H 4H 5S 6C''', '''3D 4C 5H 6H 2S''', '''Tie'''),
    ('''2S 3H 4H 5S 6C''', '''AH AC 5H 6H AS''', '''Win'''),
    ('''2S 2H 4H 5S 4C''', '''AH AC 5H 6H AS''', '''Loss'''),
    ('''2S 2H 4H 5S 4C''', '''AH AC 5H 6H 7S''', '''Win'''),
    ('''6S AD 7H 4S AS''', '''AH AC 5H 6H 7S''', '''Loss'''),
    ('''2S AH 4H 5S KC''', '''AH AC 5H 6H 7S''', '''Loss'''),
    ('''2S 3H 6H 7S 9C''', '''7H 3C TH 6H 9S''', '''Loss'''),
    ('''4S 5H 6H TS AC''', '''3S 5H 6H TS AC''', '''Win'''),
    ('''2S AH 4H 5S 6C''', '''AD 4C 5H 6H 2C''', '''Tie'''),
    ('''AS AH 3H AD AC''', '''AS AH 2H AD AC''', '''Win'''),
    ('''AH AC 5H 5C QS''', '''AH AC 5H 5C KS''', '''Loss'''),
    ('''AH AC 5H 5C QS''', '''KH KC 5H 5C QS''', '''Win'''),
    ('''7C 7S KH 2H 7H''', '''3C 3S AH 2H 3H''', '''Win'''),
    ('''3C 3S AH 2H 3H''', '''7C 7S KH 2H 7H''', '''Loss'''),
    ('''6H 5H 4H 3H 2H''', '''5H 4H 3H 2H AH''', '''Win'''),
    ('''5H 4H 3H 2H AH''', '''5H 4H 3H 2H AH''', '''Tie'''),
    ('''5H 4H 3H 2H AH''', '''6H 5H 4H 3H 2H''', '''Loss'''),
    ('''AH AD KS KC AC''', '''AH KD KH AC KC''', '''Win'''),
    ('''2H 4D 3C AS 5S''', '''2H 4D 3C 6S 5S''', '''Loss'''),
    ('''2H 3S 3C 3H 2S''', '''3S 3C 2S 2H 2D''', '''Win'''),
    ('''4D 6D 5D 2D JH''', '''3S 8S 3H TC KH''', '''Loss'''),
    ('''4S 6C 8S 3S 7S''', '''AD KS 2D 7D 7C''', '''Loss'''),
    ('''6S 4C 7H 8C 3H''', '''5H JC AH 9D 9C''', '''Loss'''),
    ('''9D 9H JH TC QH''', '''3C 2S JS 5C 7H''', '''Win'''),
    ('''2H TC 8S AD 9S''', '''4H TS 7H 2C 5C''', '''Win'''),
    ('''9D 3S 2C 7S 7C''', '''JC TD 3C TC 9H''', '''Loss'''),
)

_SCREAMING_SNAKE_CASE					=    (
    ('''2H 3H 4H 5H 6H''', True),
    ('''AS AH 2H AD AC''', False),
    ('''2H 3H 5H 6H 7H''', True),
    ('''KS AS TS QS JS''', True),
    ('''8H 9H QS JS TH''', False),
    ('''AS 3S 4S 8S 2S''', True),
)

_SCREAMING_SNAKE_CASE					=    (
    ('''2H 3H 4H 5H 6H''', True),
    ('''AS AH 2H AD AC''', False),
    ('''2H 3H 5H 6H 7H''', False),
    ('''KS AS TS QS JS''', True),
    ('''8H 9H QS JS TH''', True),
)

_SCREAMING_SNAKE_CASE					=    (
    ('''2H 4D 3C AS 5S''', True, [5, 4, 3, 2, 1_4]),
    ('''2H 5D 3C AS 5S''', False, [1_4, 5, 5, 3, 2]),
    ('''JH QD KC AS TS''', False, [1_4, 1_3, 1_2, 1_1, 1_0]),
    ('''9D 3S 2C 7S 7C''', False, [9, 7, 7, 3, 2]),
)

_SCREAMING_SNAKE_CASE					=    (
    ('''JH AH TH KH QH''', 0),
    ('''JH 9H TH KH QH''', 0),
    ('''JC KH JS JD JH''', 7),
    ('''KH KC 3S 3H 3D''', 6),
    ('''8C 9C 5C 3C TC''', 0),
    ('''JS QS 9H TS KH''', 0),
    ('''7C 7S KH 2H 7H''', 3),
    ('''3C KH 5D 5S KH''', 2),
    ('''QH 8H KD JH 8S''', 1),
    ('''2D 6D 9D TH 7D''', 0),
)

_SCREAMING_SNAKE_CASE					=    (
    ('''JH AH TH KH QH''', 2_3),
    ('''JH 9H TH KH QH''', 2_2),
    ('''JC KH JS JD JH''', 2_1),
    ('''KH KC 3S 3H 3D''', 2_0),
    ('''8C 9C 5C 3C TC''', 1_9),
    ('''JS QS 9H TS KH''', 1_8),
    ('''7C 7S KH 2H 7H''', 1_7),
    ('''3C KH 5D 5S KH''', 1_6),
    ('''QH 8H KD JH 8S''', 1_5),
    ('''2D 6D 9D TH 7D''', 1_4),
)




def    _lowerCAmelCase						(				):
			__lowercase						,		__lowercase										=							randrange(len(lowerCamelCase_				)				), randrange(len(lowerCamelCase_				)				)
			__lowercase										=							['''Loss''', '''Tie''', '''Win'''][(play >= oppo) + (play > oppo)]
			__lowercase						,		__lowercase										=							SORTED_HANDS[play], SORTED_HANDS[oppo]
			return hand, other, expected




def    _lowerCAmelCase						(				lowerCamelCase_			:  int = 1_0_0				):
			return (generate_random_hand() for _ in range(lowerCamelCase_				))




@pytest.mark.parametrize('''hand, expected'''		,       lowerCamelCase_				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  int		,       lowerCamelCase_			:  Tuple				):
			assert PokerHand(lowerCamelCase_				)._is_flush() == expected




@pytest.mark.parametrize('''hand, expected'''		,       lowerCamelCase_				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  List[str]		,       lowerCamelCase_			:  Tuple				):
			assert PokerHand(lowerCamelCase_				)._is_straight() == expected




@pytest.mark.parametrize('''hand, expected, card_values'''		,       lowerCamelCase_				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  Tuple		,       lowerCamelCase_			:  Any		,       lowerCamelCase_			:  Union[str, Any]				):
			__lowercase										=							PokerHand(lowerCamelCase_				)
			assert player._is_five_high_straight() == expected
			assert player._card_values == card_values




@pytest.mark.parametrize('''hand, expected'''		,       lowerCamelCase_				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  Any		,       lowerCamelCase_			:  List[Any]				):
			assert PokerHand(lowerCamelCase_				)._is_same_kind() == expected




@pytest.mark.parametrize('''hand, expected'''		,       lowerCamelCase_				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  str		,       lowerCamelCase_			:  Any				):
			assert PokerHand(lowerCamelCase_				)._hand_type == expected




@pytest.mark.parametrize('''hand, other, expected'''		,       lowerCamelCase_				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  Optional[Any]		,       lowerCamelCase_			:  List[str]		,       lowerCamelCase_			:  Union[str, Any]				):
			assert PokerHand(lowerCamelCase_				).compare_with(PokerHand(lowerCamelCase_				)				) == expected




@pytest.mark.parametrize('''hand, other, expected'''		,       generate_random_hands()				)
def    _lowerCAmelCase						(				lowerCamelCase_			:  Any		,       lowerCamelCase_			:  str		,       lowerCamelCase_			:  List[Any]				):
			assert PokerHand(lowerCamelCase_				).compare_with(PokerHand(lowerCamelCase_				)				) == expected




def    _lowerCAmelCase						(				):
			__lowercase										=							[PokerHand(lowerCamelCase_				) for hand in SORTED_HANDS]
			__lowercase										=							poker_hands.copy()
			shuffle(lowerCamelCase_				)
			__lowercase										=							chain(sorted(lowerCamelCase_				)				)
			for index, hand in enumerate(lowerCamelCase_				):
						assert hand == poker_hands[index]




def    _lowerCAmelCase						(				):
			# Test that five high straights are compared correctly.
			__lowercase										=							[PokerHand('''2D AC 3H 4H 5S'''				), PokerHand('''2S 3H 4H 5S 6C'''				)]
			pokerhands.sort(reverse=lowerCamelCase_				)
			assert pokerhands[0].__str__() == "2S 3H 4H 5S 6C"




def    _lowerCAmelCase						(				):
			# Multiple calls to five_high_straight function should still return True
			# and shouldn't mutate the list in every call other than the first.
			__lowercase										=							PokerHand('''2C 4S AS 3D 5C'''				)
			__lowercase										=							True
			__lowercase										=							[5, 4, 3, 2, 1_4]
			for _ in range(1_0				):
						assert pokerhand._is_five_high_straight() == expected
						assert pokerhand._card_values == expected_card_values




def    _lowerCAmelCase						(				):
			# Problem number 54 from Project Euler
			# Testing from poker_hands.txt file
			__lowercase										=							0
			__lowercase										=							os.path.abspath(os.path.dirname(lowerCamelCase_				)				)
			__lowercase										=							os.path.join(lowerCamelCase_		,       '''poker_hands.txt'''				)
			with open(lowerCamelCase_				) as file_hand:
						for line in file_hand:
									__lowercase										=							line[:1_4].strip()
									__lowercase										=							line[1_5:].strip()
									__lowercase						,		__lowercase										=							PokerHand(lowerCamelCase_				), PokerHand(lowerCamelCase_				)
									__lowercase										=							player.compare_with(lowerCamelCase_				)
									if output == "Win":
												answer += 1
			assert answer == 3_7_6












'''simple docstring'''



import argparse
from pathlib import Path

import torch

from transformers import OPTConfig, OPTModel
from transformers.utils import logging


logging.set_verbosity_info()
_SCREAMING_SNAKE_CASE					=    logging.get_logger(__name__)




def    _lowerCAmelCase						(				lowerCamelCase_			:  int				):
			__lowercase										=							torch.load(lowerCamelCase_		,       map_location='''cpu'''				)
			if "model" in sd.keys():
						__lowercase										=							torch.load(lowerCamelCase_		,       map_location='''cpu'''				)['''model''']

			# pop unnecessary weights
			__lowercase										=							[
			    '''decoder.version''',
			    '''decoder.output_projection.weight''',
			]
			for key in keys_to_delete:
						if key in sd:
									sd.pop(lowerCamelCase_				)

			__lowercase										=							{
			    '''decoder.project_in_dim.weight''': '''decoder.project_in.weight''',
			    '''decoder.project_out_dim.weight''': '''decoder.project_out.weight''',
			    '''decoder.layer_norm.weight''': '''decoder.final_layer_norm.weight''',
			    '''decoder.layer_norm.bias''': '''decoder.final_layer_norm.bias''',
			}
			for old_key, new_key in keys_to_rename.items():
						if old_key in sd:
									__lowercase										=							sd.pop(lowerCamelCase_				)

			__lowercase										=							list(sd.keys()				)
			for key in keys:
						if ".qkv_proj." in key:
									__lowercase										=							sd[key]
									# We split QKV in separate Q,K,V

									__lowercase										=							key.replace('''.qkv_proj.'''		,       '''.q_proj.'''				)
									__lowercase										=							key.replace('''.qkv_proj.'''		,       '''.k_proj.'''				)
									__lowercase										=							key.replace('''.qkv_proj.'''		,       '''.v_proj.'''				)

									__lowercase										=							value.shape[0]
									assert depth % 3 == 0
									# `SequeuceParallelTransformerBlock` has QKV weight is separated in K,V,Q despite the naming:
									# https://cs.github.com/facebookresearch/metaseq/blob/51871bd73cd04c038f239ea2a26db1d7f6b37927/metaseq/modules/sequence_parallel_transformer_layer.py#L97
									__lowercase						,		__lowercase						,		__lowercase										=							torch.split(lowerCamelCase_		,       depth // 3		,       dim=0				)

									__lowercase										=							q
									__lowercase										=							k
									__lowercase										=							v
									del sd[key]

			return sd




@torch.no_grad()
def    _lowerCAmelCase						(				lowerCamelCase_			:  Union[str, Any]		,       lowerCamelCase_			:  Tuple		,       lowerCamelCase_			:  Union[str, Any]=None				):
			__lowercase										=							load_checkpoint(lowerCamelCase_				)

			if config is not None:
						__lowercase										=							OPTConfig.from_pretrained(lowerCamelCase_				)
			else:
						__lowercase										=							OPTConfig()

			__lowercase										=							OPTModel(lowerCamelCase_				).half().eval()
			model.load_state_dict(lowerCamelCase_				)

			# Check results
			Path(lowerCamelCase_				).mkdir(exist_ok=lowerCamelCase_				)
			model.save_pretrained(lowerCamelCase_				)


if __name__ == "__main__":
	_SCREAMING_SNAKE_CASE					=    argparse.ArgumentParser()
	# Required parameters
	parser.add_argument(
	    '''--fairseq_path''',
	    type=str,
	    help=(
	        '''path to fairseq checkpoint in correct format. You can find all checkpoints in the correct format here:'''
	        ''' https://huggingface.co/models?other=opt_metasq'''
	    ),
	)
	parser.add_argument('''--pytorch_dump_folder_path''', default=None, type=str, help='''Path to the output PyTorch model.''')
	parser.add_argument('''--hf_config''', default=None, type=str, help='''Define HF config.''')
	_SCREAMING_SNAKE_CASE					=    parser.parse_args()
	convert_opt_checkpoint(args.fairseq_path, args.pytorch_dump_folder_path, config=args.hf_config)





