Automated Code Change

PiperOrigin-RevId: 663537270
tensorflow · Aug 16, 2024 · 215e9cf · 215e9cf
1 parent 7314fd9
commit 215e9cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_probability/python/internal/backend/numpy/ops.py b/tensorflow_probability/python/internal/backend/numpy/ops.py
@@ -311,7 +311,7 @@ def _default_convert_to_tensor_with_dtype(value, dtype,
                                   or np.issubdtype(dtype, np.floating)):
       raise MixedTypesError()
     if dtype == np.int32 and _is_int64(value):
-      raise _Int64ToInt32Error(np.array(value, dtype=dtype))
+      raise _Int64ToInt32Error(np.array(value).astype(dtype))
     if dtype == np.bool_:
       # Can't downcast an int to a bool
       raise TypeConversionError(value, dtype)