MERGE MAIN

blackjax/mcmc/integrators.py

@@ -37,9 +37,6 @@
     "isokinetic_omelyan",
     "isokinetic_yoshida",
     "implicit_midpoint",
-    "calls_per_integrator_step",
-    "name_integrator",
-    "integrator_order",
 ]
 
 
@@ -402,48 +399,6 @@ def isokinetic_integrator(
 isokinetic_omelyan = generate_isokinetic_integrator(omelyan_coefficients)
 
 
-def calls_per_integrator_step(c):
-    if c == velocity_verlet_coefficients:
-        return 1
-    if c == mclachlan_coefficients:
-        return 2
-    if c == yoshida_coefficients:
-        return 3
-    if c == omelyan_coefficients:
-        return 5
-
-    else:
-        raise Exception("No such integrator exists in blackjax")
-
-
-def name_integrator(c):
-    if c == velocity_verlet_coefficients:
-        return "velocity_verlet"
-    if c == mclachlan_coefficients:
-        return "mclachlan"
-    if c == yoshida_coefficients:
-        return "yoshida"
-    if c == omelyan_coefficients:
-        return "omelyan"
-
-    else:
-        raise Exception("No such integrator exists in blackjax")
-
-
-def integrator_order(c):
-    if c == velocity_verlet_coefficients:
-        return 2
-    if c == mclachlan_coefficients:
-        return 2
-    if c == yoshida_coefficients:
-        return 4
-    if c == omelyan_coefficients:
-        return 4
-
-    else:
-        raise Exception("No such integrator exists in blackjax")
-
-
 def partially_refresh_momentum(momentum, rng_key, step_size, L):
     """Adds a small noise to momentum and normalizes.