




"""simple docstring"""





from random import shuffle

import tensorflow as tf
from numpy import array

def        _UpperCamelCase     (      UpperCamelCase					,							UpperCamelCase				)      ->      List[str]:


			"""simple docstring"""
			__UpperCAmelCase						:			str												=       int(UpperCAmelCase_				)
			assert noofclusters < len(UpperCAmelCase_				)

			# Find out the dimensionality
			__UpperCAmelCase						:			Any												=       len(vectors[0]				)

			# Will help select random centroids from among the available vectors
			__UpperCAmelCase						:			Union[str, Any]												=       list(range(len(UpperCAmelCase_				)				)				)
			shuffle(UpperCAmelCase_				)

			# GRAPH OF COMPUTATION
			# We initialize a new graph and set it as the default during each run
			# of this algorithm. This ensures that as this function is called
			# multiple times, the default graph doesn't keep getting crowded with
			# unused ops and Variables from previous function calls.

			__UpperCAmelCase						:			List[Any]												=       tf.Graph()

			with graph.as_default():
						# SESSION OF COMPUTATION

						__UpperCAmelCase						:			str												=       tf.Session()

						##CONSTRUCTING THE ELEMENTS OF COMPUTATION

						##First lets ensure we have a Variable vector for each centroid,
						##initialized to one of the vectors from the available data points
						__UpperCAmelCase						:			Any												=       [
						    tf.Variable(vectors[vector_indices[i]]				) for i in range(UpperCAmelCase_				)
						]
						##These nodes will assign the centroid Variables the appropriate
						##values
						__UpperCAmelCase						:			int												=       tf.placeholder("float64"					,							[dim]				)
						__UpperCAmelCase						:			Union[str, Any]												=       []
						for centroid in centroids:
									cent_assigns.append(tf.assign(UpperCAmelCase_					,							UpperCAmelCase_				)				)

						##Variables for cluster assignments of individual vectors(initialized
						##to 0 at first)
						__UpperCAmelCase						:			Dict												=       [tf.Variable(0				) for i in range(len(UpperCAmelCase_				)				)]
						##These nodes will assign an assignment Variable the appropriate
						##value
						__UpperCAmelCase						:			int												=       tf.placeholder("int32"				)
						__UpperCAmelCase						:			List[Any]												=       []
						for assignment in assignments:
									cluster_assigns.append(tf.assign(UpperCAmelCase_					,							UpperCAmelCase_				)				)

						##Now lets construct the node that will compute the mean
						# The placeholder for the input
						__UpperCAmelCase						:			List[str]												=       tf.placeholder("float"					,							[None, dim]				)
						# The Node/op takes the input and computes a mean along the 0th
						# dimension, i.e. the list of input vectors
						__UpperCAmelCase						:			int												=       tf.reduce_mean(UpperCAmelCase_					,							0				)

						##Node for computing Euclidean distances
						# Placeholders for input
						__UpperCAmelCase						:			List[str]												=       tf.placeholder("float"					,							[dim]				)
						__UpperCAmelCase						:			List[Any]												=       tf.placeholder("float"					,							[dim]				)
						__UpperCAmelCase						:			Dict												=       tf.sqrt(tf.reduce_sum(tf.pow(tf.sub(UpperCAmelCase_					,							UpperCAmelCase_				)					,							2				)				)				)

						##This node will figure out which cluster to assign a vector to,
						##based on Euclidean distances of the vector from the centroids.
						# Placeholder for input
						__UpperCAmelCase						:			List[Any]												=       tf.placeholder("float"					,							[noofclusters]				)
						__UpperCAmelCase						:			Tuple												=       tf.argmin(UpperCAmelCase_					,							0				)

						##INITIALIZING STATE VARIABLES

						##This will help initialization of all Variables defined with respect
						##to the graph. The Variable-initializer should be defined after
						##all the Variables have been constructed, so that each of them
						##will be included in the initialization.
						__UpperCAmelCase						:			Any												=       tf.initialize_all_variables()

						# Initialize all variables
						sess.run(UpperCAmelCase_				)

						##CLUSTERING ITERATIONS

						# Now perform the Expectation-Maximization steps of K-Means clustering
						# iterations. To keep things simple, we will only do a set number of
						# iterations, instead of using a Stopping Criterion.
						__UpperCAmelCase						:			List[Any]												=       100
						for _ in range(UpperCAmelCase_				):
									##EXPECTATION STEP
									##Based on the centroid locations till last iteration, compute
									##the _expected_ centroid assignments.
									# Iterate over each vector
									for vector_n in range(len(UpperCAmelCase_				)				):
												__UpperCAmelCase						:			int												=       vectors[vector_n]
												# Compute Euclidean distance between this vector and each
												# centroid. Remember that this list cannot be named
												#'centroid_distances', since that is the input to the
												# cluster assignment node.
												__UpperCAmelCase						:			List[str]												=       [
												    sess.run(UpperCAmelCase_					,							feed_dict={va: vect, va: sess.run(UpperCAmelCase_				)}				)
												    for centroid in centroids
												]
												# Now use the cluster assignment node, with the distances
												# as the input
												__UpperCAmelCase						:			Dict												=       sess.run(
												    UpperCAmelCase_					,							feed_dict={centroid_distances: distances}				)
												# Now assign the value to the appropriate state variable
												sess.run(
												    cluster_assigns[vector_n]					,							feed_dict={assignment_value: assignment}				)

									##MAXIMIZATION STEP
									# Based on the expected state computed from the Expectation Step,
									# compute the locations of the centroids so as to maximize the
									# overall objective of minimizing within-cluster Sum-of-Squares
									for cluster_n in range(UpperCAmelCase_				):
												# Collect all the vectors assigned to this cluster
												__UpperCAmelCase						:			Optional[Any]												=       [
												    vectors[i]
												    for i in range(len(UpperCAmelCase_				)				)
												    if sess.run(assignments[i]				) == cluster_n
												]
												# Compute new centroid location
												__UpperCAmelCase						:			Tuple												=       sess.run(
												    UpperCAmelCase_					,							feed_dict={mean_input: array(UpperCAmelCase_				)}				)
												# Assign value to appropriate variable
												sess.run(
												    cent_assigns[cluster_n]					,							feed_dict={centroid_value: new_location}				)

        # Return centroids and assignments
						__UpperCAmelCase						:			str												=       sess.run(UpperCAmelCase_				)
						__UpperCAmelCase						:			Any												=       sess.run(UpperCAmelCase_				)
						return centroids, assignments






'''simple docstring'''



from __future__ import annotations

import unittest

from transformers import XGLMConfig, XGLMTokenizer, is_tf_available
from transformers.testing_utils import require_tf, slow

from ...test_configuration_common import ConfigTester
from ...test_modeling_tf_common import TFModelTesterMixin, floats_tensor, ids_tensor, random_attention_mask
from ...test_pipeline_mixin import PipelineTesterMixin


if is_tf_available():
      import tensorflow as tf

      from transformers.models.xglm.modeling_tf_xglm import (
          TF_XGLM_PRETRAINED_MODEL_ARCHIVE_LIST,
          TFXGLMForCausalLM,
          TFXGLMModel,
      )





@require_tf
class   UpperCAmelCase_    :





    """simple docstring"""






    lowerCamelCase    :   Dict					=     XGLMConfig
    lowerCamelCase    :   List[str]					=     {}
    lowerCamelCase    :   Union[str, Any]					=     'gelu'







    def __init__(				self     ,	SCREAMING_SNAKE_CASE_     ,	SCREAMING_SNAKE_CASE_=14     ,	SCREAMING_SNAKE_CASE_=7     ,	SCREAMING_SNAKE_CASE_=True     ,	SCREAMING_SNAKE_CASE_=True     ,	SCREAMING_SNAKE_CASE_=True     ,	SCREAMING_SNAKE_CASE_=99     ,	SCREAMING_SNAKE_CASE_=32     ,	SCREAMING_SNAKE_CASE_=2     ,	SCREAMING_SNAKE_CASE_=4     ,	SCREAMING_SNAKE_CASE_=37     ,	SCREAMING_SNAKE_CASE_="gelu"     ,	SCREAMING_SNAKE_CASE_=0.1     ,	SCREAMING_SNAKE_CASE_=0.1     ,	SCREAMING_SNAKE_CASE_=5_12     ,	SCREAMING_SNAKE_CASE_=0.0_2     ,	)							->	Any:
         __lowerCamelCase  :      int														=						parent
         __lowerCamelCase  :      Optional[int]														=						batch_size
         __lowerCamelCase  :      Optional[Any]														=						seq_length
         __lowerCamelCase  :      Optional[int]														=						is_training
         __lowerCamelCase  :      str														=						use_input_mask
         __lowerCamelCase  :      Dict														=						use_labels
         __lowerCamelCase  :      Union[str, Any]														=						vocab_size
         __lowerCamelCase  :      List[Any]														=						d_model
         __lowerCamelCase  :      List[Any]														=						num_hidden_layers
         __lowerCamelCase  :      List[Any]														=						num_attention_heads
         __lowerCamelCase  :      Optional[Any]														=						ffn_dim
         __lowerCamelCase  :      List[Any]														=						activation_function
         __lowerCamelCase  :      List[Any]														=						activation_dropout
         __lowerCamelCase  :      List[Any]														=						attention_dropout
         __lowerCamelCase  :      Union[str, Any]														=						max_position_embeddings
         __lowerCamelCase  :      Tuple														=						initializer_range
         __lowerCamelCase  :      int														=						None
         __lowerCamelCase  :      int														=						0
         __lowerCamelCase  :      Tuple														=						2
         __lowerCamelCase  :      Tuple														=						1







    def  lowercase_							(				self      )							->	Any:
         return XGLMConfig.from_pretrained('facebook/xglm-564M'      )







    def  lowercase_							(				self      )							->	Tuple:
         __lowerCamelCase  :      Optional[Any]														=						tf.clip_by_value(
             ids_tensor([self.batch_size, self.seq_length]     ,	self.vocab_size      )     ,	clip_value_min=0     ,	clip_value_max=3      )

         __lowerCamelCase  :      Optional[int]														=						None
         if self.use_input_mask:
              __lowerCamelCase  :      Any														=						random_attention_mask([self.batch_size, self.seq_length]      )

         __lowerCamelCase  :      str														=						self.get_config()

         __lowerCamelCase  :      List[Any]														=						floats_tensor([self.num_hidden_layers, self.num_attention_heads]     ,	2      )

         return (
             config,
             input_ids,
             input_mask,
             head_mask,
         )







    def  lowercase_							(				self      )							->	Optional[int]:
         return XGLMConfig(
             vocab_size=self.vocab_size     ,	d_model=self.hidden_size     ,	num_layers=self.num_hidden_layers     ,	attention_heads=self.num_attention_heads     ,	ffn_dim=self.ffn_dim     ,	activation_function=self.activation_function     ,	activation_dropout=self.activation_dropout     ,	attention_dropout=self.attention_dropout     ,	max_position_embeddings=self.max_position_embeddings     ,	initializer_range=self.initializer_range     ,	use_cache=SCREAMING_SNAKE_CASE_     ,	bos_token_id=self.bos_token_id     ,	eos_token_id=self.eos_token_id     ,	pad_token_id=self.pad_token_id     ,	return_dict=SCREAMING_SNAKE_CASE_     ,	)







    def  lowercase_							(				self      )							->	str:
         __lowerCamelCase  :      List[Any]														=						self.prepare_config_and_inputs()

         (
             (
             __lowerCamelCase
         )							,			(
             __lowerCamelCase
         )							,			(
             __lowerCamelCase
         )							,			(
             __lowerCamelCase
         )							,			
         )  :      str														=						config_and_inputs

         __lowerCamelCase  :      Union[str, Any]														=						{
             'input_ids': input_ids,
             'head_mask': head_mask,
         }

         return config, inputs_dict







@require_tf
class   UpperCAmelCase_    (_UpperCAmelCase   ,						_UpperCAmelCase   ,						unittest.TestCase    ):





    """simple docstring"""






    lowerCamelCase    :   Optional[Any]					=     (TFXGLMModel, TFXGLMForCausalLM) if is_tf_available() else ()
    lowerCamelCase    :   List[Any]					=     (TFXGLMForCausalLM,) if is_tf_available() else ()
    lowerCamelCase    :   Any					=     (
        {'feature-extraction': TFXGLMModel, 'text-generation': TFXGLMForCausalLM} if is_tf_available() else {}
    )
    lowerCamelCase    :   List[Any]					=     False
    lowerCamelCase    :   Dict					=     False
    lowerCamelCase    :   Union[str, Any]					=     False







    def  lowercase_							(				self      )							->	List[Any]:
         __lowerCamelCase  :      str														=						TFXGLMModelTester(self      )
         __lowerCamelCase  :      List[Any]														=						ConfigTester(self     ,	config_class=SCREAMING_SNAKE_CASE_     ,	n_embd=37      )







    def  lowercase_							(				self      )							->	Dict:
         self.config_tester.run_common_tests()







    @slow
    def  lowercase_							(				self      )							->	Optional[int]:
         for model_name in TF_XGLM_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
              __lowerCamelCase  :      Optional[Any]														=						TFXGLMModel.from_pretrained(SCREAMING_SNAKE_CASE_      )
              self.assertIsNotNone(SCREAMING_SNAKE_CASE_      )







    @unittest.skip(reason='Currently, model embeddings are going to undergo a major refactor.'      )
    def  lowercase_							(				self      )							->	Any:
         super().test_resize_token_embeddings()







@require_tf
class   UpperCAmelCase_    (unittest.TestCase    ):





    """simple docstring"""






    @slow
    def  lowercase_							(				self     ,	SCREAMING_SNAKE_CASE_=True      )							->	List[str]:
         __lowerCamelCase  :      Optional[Any]														=						TFXGLMForCausalLM.from_pretrained('facebook/xglm-564M'      )
         __lowerCamelCase  :      int														=						tf.convert_to_tensor([[2, 2_68, 98_65]]     ,	dtype=tf.intaa      )  # The dog
         # </s> The dog is a very friendly dog. He is very affectionate and loves to play with other
         # fmt: off
         __lowerCamelCase  :      Optional[int]														=						[2, 2_68, 98_65, 67, 11, 19_88, 5_72_52, 98_65, 5, 9_84, 67, 19_88, 21_38_38, 16_58, 53, 7_04_46, 33, 66_57, 2_78, 15_81]
         # fmt: on
         __lowerCamelCase  :      Any														=						model.generate(SCREAMING_SNAKE_CASE_     ,	do_sample=SCREAMING_SNAKE_CASE_     ,	num_beams=1      )
         if verify_outputs:
              self.assertListEqual(output_ids[0].numpy().tolist()     ,	SCREAMING_SNAKE_CASE_      )







    @slow
    def  lowercase_							(				self      )							->	int:
         __lowerCamelCase  :      Any														=						XGLMTokenizer.from_pretrained('facebook/xglm-564M'      )
         __lowerCamelCase  :      Tuple														=						TFXGLMForCausalLM.from_pretrained('facebook/xglm-564M'      )

         tf.random.set_seed(0      )
         __lowerCamelCase  :      List[Any]														=						tokenizer('Today is a nice day and'     ,	return_tensors='tf'      )
         __lowerCamelCase  :      int														=						tokenized.input_ids
         # forces the generation to happen on CPU, to avoid GPU-related quirks (and assure same output regardless of the available devices)
         with tf.device(':/CPU:0'      ):
              __lowerCamelCase  :      Tuple														=						model.generate(SCREAMING_SNAKE_CASE_     ,	do_sample=SCREAMING_SNAKE_CASE_     ,	seed=[7, 0]      )
         __lowerCamelCase  :      Optional[Any]														=						tokenizer.decode(output_ids[0]     ,	skip_special_tokens=SCREAMING_SNAKE_CASE_      )

         __lowerCamelCase  :      Union[str, Any]														=						(
             'Today is a nice day and warm evening here over Southern Alberta!! Today when they closed schools due'
         )
         self.assertEqual(SCREAMING_SNAKE_CASE_     ,	SCREAMING_SNAKE_CASE_      )







    @slow
    def  lowercase_							(				self      )							->	int:
         __lowerCamelCase  :      Tuple														=						TFXGLMForCausalLM.from_pretrained('facebook/xglm-564M'      )
         __lowerCamelCase  :      Any														=						XGLMTokenizer.from_pretrained('facebook/xglm-564M'      )

         __lowerCamelCase  :      Any														=						'left'

         # use different length sentences to test batching
         __lowerCamelCase  :      Any														=						[
             'This is an extremelly long sentence that only exists to test the ability of the model to cope with '
             'left-padding, such as in batched generation. The output for the sequence below should be the same '
             'regardless of whether left padding is applied or not. When',
             'Hello, my dog is a little',
         ]

         __lowerCamelCase  :      Any														=						tokenizer(SCREAMING_SNAKE_CASE_     ,	return_tensors='tf'     ,	padding=SCREAMING_SNAKE_CASE_      )
         __lowerCamelCase  :      Optional[int]														=						inputs['input_ids']

         __lowerCamelCase  :      str														=						model.generate(input_ids=SCREAMING_SNAKE_CASE_     ,	attention_mask=inputs['attention_mask']     ,	max_new_tokens=12      )

         __lowerCamelCase  :      Optional[int]														=						tokenizer(sentences[0]     ,	return_tensors='tf'      ).input_ids
         __lowerCamelCase  :      int														=						model.generate(input_ids=SCREAMING_SNAKE_CASE_     ,	max_new_tokens=12      )

         __lowerCamelCase  :      Optional[Any]														=						tokenizer(sentences[1]     ,	return_tensors='tf'      ).input_ids
         __lowerCamelCase  :      Optional[Any]														=						model.generate(input_ids=SCREAMING_SNAKE_CASE_     ,	max_new_tokens=12      )

         __lowerCamelCase  :      Union[str, Any]														=						tokenizer.batch_decode(SCREAMING_SNAKE_CASE_     ,	skip_special_tokens=SCREAMING_SNAKE_CASE_      )
         __lowerCamelCase  :      Union[str, Any]														=						tokenizer.decode(output_non_padded[0]     ,	skip_special_tokens=SCREAMING_SNAKE_CASE_      )
         __lowerCamelCase  :      Union[str, Any]														=						tokenizer.decode(output_padded[0]     ,	skip_special_tokens=SCREAMING_SNAKE_CASE_      )

         __lowerCamelCase  :      Optional[int]														=						[
             'This is an extremelly long sentence that only exists to test the ability of the model to cope with '
             'left-padding, such as in batched generation. The output for the sequence below should be the same '
             'regardless of whether left padding is applied or not. When left padding is applied, the sequence will be '
             'a single',
             'Hello, my dog is a little bit of a shy one, but he is very friendly',
         ]
         self.assertListEqual(SCREAMING_SNAKE_CASE_     ,	SCREAMING_SNAKE_CASE_      )
         self.assertListEqual(SCREAMING_SNAKE_CASE_     ,	[non_padded_sentence, padded_sentence]      )






