added todo

ott-jax · Nov 6, 2023 · ae54f1e · ae54f1e
1 parent b4777a1
commit ae54f1e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/solvers/linear/univariate_test.py b/tests/solvers/linear/univariate_test.py
@@ -87,6 +87,8 @@ def test_cdf_grad(
       self,
       rng: jax.Array,
   ):
+    # TODO: Once a `check_grad` function is implemented, replace the code
+    # blocks before with `check_grad`'s.
     cost_fn = costs.SqEuclidean()
     rngs = jax.random.split(rng, 4)
     eps, tol = 1e-4, 1e-3