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

Fix the jax2tf failure in mypy: https://github.com/google/jax/actions/runs/5094063162/jobs/9157426652?pr=16155

Browse Source 
PiperOrigin-RevId: 535692853
This commit is contained in:
Yash Katariya 2023-05-26 12:57:01 -07:00 committed by jax authors
parent fe3fed3627
commit d62bc0f795
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/experimental/jax2tf/tests/jax_export_test.py
View File

@ -26,7 +26,7 @@ from jax.experimental.jax2tf import jax_export
try:
from jax.experimental.jax2tf import jax2tf # TODO: temporary
except ImportError:
jax2tf = None
jax2tf = None # type: ignore
from jax._src import core
from jax._src import test_util as jtu