RESOLVE MYPY ISSUE

blackjax-devs · May 27, 2024 · 40b5b98 · 40b5b98
1 parent f4e8064
commit 40b5b98
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 5 deletions.
diff --git a/blackjax/adaptation/mclmc_adaptation.py b/blackjax/adaptation/mclmc_adaptation.py
@@ -550,10 +550,6 @@ def L_step_size_adaptation(state, params, num_steps, rng_key):
             if diagonal_preconditioning:
                 params = params._replace(sqrt_diag_cov=jnp.sqrt(variances))
 
-                # state = jax.lax.scan(step, init= state, xs= jnp.ones(steps), length= steps)[0]
-                # dyn, _, hyp, adap, kalman_state = state
-                # TODO ^
-
             initial_da, update_da, final_da = dual_averaging_adaptation(target=target)
             (
                 (state, params, (dual_avg_state, step_size_max), (_, average)),

diff --git a/blackjax/mcmc/adjusted_mclmc.py b/blackjax/mcmc/adjusted_mclmc.py
@@ -33,7 +33,6 @@ def init(position: ArrayLikeTree, logdensity_fn: Callable, random_generator_arg:
     return DynamicHMCState(position, logdensity, logdensity_grad, random_generator_arg)
 
 
-# TODO: no default for sqrt_diag_cov
 def build_kernel(
     integration_steps_fn,
     integrator: Callable = integrators.isokinetic_mclachlan,