mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid using os.linesep

Browse Source
This commit is contained in:
Cloud Han 2023-06-16 00:05:32 +08:00
parent 48ad9a6f3e
commit 7b8bdfbe0f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
jax/_src/lax/control_flow/common.py
View File

@ -41,6 +41,9 @@ def _typecheck_param(prim, param, name, msg_required, pred):
msg = (f'invalid {prim} param {name} of type {type(param).__name__}, '
f'{msg_required} required:')
param_str = str(param)
# Avoid using os.linesep here to have the same multi-line error message
# format on different platforms.
sep = '\n' if '\n' in param_str or '\r' in param_str else ' '
sep = os.linesep if os.linesep in param_str else ' '
msg = sep.join([msg, param_str])
raise core.JaxprTypeError(msg)