code: def testTruncationRewardDTypeTensor(self):
      observation = tf.random.uniform([5, 2])
      reward = tf.random.uniform([2], dtype=tf.float64)
      discount = 1.0
      time_step = ts.truncation(observation, reward, discount)
  
      self.assertEqual(time_step.reward.dtype, reward.dtype)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import sys
  import numpy
  import tensorflow
  from tf_agents.specs import array_spec
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
  from tf_agents.utils import nest_utils
label_desc: Use image node_id 9 for calling the TimeStepTest obj's underlying member method code with example usage: obj.testTruncationRewardDTypeTensor() without return types