code: def test_bahdanau_not_normalized():
      set_random_state_for_tf_and_np()
      policy = tf.keras.mixed_precision.global_policy()
      create_attention_mechanism = wrapper.BahdanauAttention
      create_attention_kwargs = {"kernel_initializer": "ones"}
      expected_final_output = basic_decoder.BasicDecoderOutput(
          rnn_output=ResultSummary(
              shape=(5, 3, 6),
              dtype=policy.compute_dtype,
              mean=-0.003204414,
          ),
          sample_id=ResultSummary(
              shape=(5, 3), dtype=np.dtype(np.int32), mean=3.2
          ),
      )
      expected_final_state = wrapper.AttentionWrapperState(
          cell_state=[
              ResultSummary(
                  shape=(5, 9),
                  dtype=policy.compute_dtype,
                  mean=0.40868404,
              ),
              ResultSummary(
                  shape=(5, 9),
                  dtype=policy.compute_dtype,
                  mean=0.89017969,
              ),
          ],
          attention=ResultSummary(
              shape=(5, 6), dtype=policy.compute_dtype, mean=0.041453815
          ),
          alignments=ResultSummary(
              shape=(5, 8), dtype=policy.compute_dtype, mean=0.125
          ),
          attention_state=ResultSummary(
              shape=(5, 8), dtype=policy.compute_dtype, mean=0.125
          ),
          alignment_history=(),
      )
      expected_final_alignment_history = ResultSummary(
          shape=(3, 5, 8), dtype=policy.compute_dtype, mean=0.125
      )
  
      _test_with_attention(
          create_attention_mechanism,
          expected_final_output,
          expected_final_state,
          alignment_history=True,
          create_query_layer=True,
          expected_final_alignment_history=expected_final_alignment_history,
          create_attention_kwargs=create_attention_kwargs,
      )
  
imports: import collections
  from packaging.version import Version
  import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.seq2seq import attention_wrapper
  from tensorflow_addons.seq2seq import basic_decoder
  from tensorflow_addons.seq2seq import sampler
label_desc: Use image node_id 16 for calling a global function with example usage: test_bahdanau_not_normalized() without return types