Skip to content

Commit ef3e3b5

Browse files
author
jax authors
committed
Rollback fix repr() of jit-compiled functions
PiperOrigin-RevId: 402621480
1 parent 7fa6b1b commit ef3e3b5

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

tests/api_test.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -80,14 +80,6 @@ def jit(self):
8080
# Tensorflow.
8181
return api._cpp_jit
8282

83-
@unittest.skipIf(jax._src.lib._xla_extension_version < 40,
84-
"Test requires jaxlib 0.1.72")
85-
def test_jit_func_repr(self):
86-
def my_function():
87-
return
88-
jitted = jit(my_function)
89-
self.assertEqual(repr(jitted), repr(my_function))
90-
9183
def test_jit_of_noncallable(self):
9284
self.assertRaisesRegex(TypeError, "Expected a callable value.*",
9385
lambda: self.jit(3))

0 commit comments

Comments
 (0)