We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7fa6b1b commit ef3e3b5Copy full SHA for ef3e3b5
tests/api_test.py
@@ -80,14 +80,6 @@ def jit(self):
80
# Tensorflow.
81
return api._cpp_jit
82
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
-
91
def test_jit_of_noncallable(self):
92
self.assertRaisesRegex(TypeError, "Expected a callable value.*",
93
lambda: self.jit(3))
0 commit comments