We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ad3ad63 commit dcd5a91Copy full SHA for dcd5a91
1 file changed
test/b/test_b_jax.py
@@ -89,7 +89,7 @@ def test_b_poly_of_degree_0(self):
89
self.assertEqual((m,), y.shape)
90
self.assertTrue(jnp.allclose(y, 2.0))
91
92
- def test_b_b_poly_of_degree_0_grad(self):
+ def test_b_poly_of_degree_0_grad(self):
93
def b_poly_sum(b, x):
94
"""To test the gradient."""
95
return jnp.sum(b_poly(b, x))
0 commit comments