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

Catch ImportError when importing tf instead of a broad exception catch. If not, this leads to weird errors in the other tests down the line.

Browse Source 
PiperOrigin-RevId: 510206006
This commit is contained in:
Yash Katariya 2023-02-16 11:57:04 -08:00 committed by jax authors
parent 54269c1145
commit 34324f80e9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/array_interoperability_test.py
View File

@ -37,7 +37,7 @@ try:
import tensorflow as tf
tf_version = tuple(
int(x) for x in tf.version.VERSION.split("-")[0].split("."))
except:
except ImportError:
tf = None