fix type promotion issue in tensorspec

PiperOrigin-RevId: 664858958 Change-Id: Ia57049212d622770c617342b17ead8c542a650f4
tensorflow · Aug 19, 2024 · 8662dab · 8662dab
1 parent 3744cbc
commit 8662dab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tf_agents/specs/tensor_spec.py b/tf_agents/specs/tensor_spec.py
@@ -293,7 +293,7 @@ def _unique_vals(vals):
     # behavior.
     # However +1 will cause overflow, in such cases we use the original maxval.
     if sampling_dtype.is_integer and maxval < sampling_dtype.max:
-      maxval = maxval + 1
+      maxval = int(maxval) + 1
 
     shape = tf.convert_to_tensor(spec.shape, dtype=tf.int32)
     full_shape = tf.concat((outer_dims, shape), axis=0)