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

Remove obsolete PIL compatibility code

Browse Source
This commit is contained in:
Jake VanderPlas 2023-04-10 15:23:10 -07:00
parent 8abe03daed
commit ad72db6184
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/image_test.py
View File

@ -14,7 +14,6 @@
from functools import partial
import unittest
import warnings
import numpy as np
@ -29,12 +28,7 @@ from jax.config import config
# We use TensorFlow and PIL as reference implementations.
try:
# TODO(jakevdp): remove this warning filter when keras requirement is updated.
# Warning comes form PIL>=9.1.0 with keras<2.10.0
with warnings.catch_warnings():
warnings.filterwarnings('ignore', category=DeprecationWarning,
message=".*is deprecated and will be removed in Pillow 10.*")
import tensorflow as tf
import tensorflow as tf
except ImportError:
tf = None