Fix rebase

Author: michaelosthege

tests/sampling/test_mcmc.py

@@ -40,11 +40,11 @@
     NUTS,
     BinaryGibbsMetropolis,
     CategoricalGibbsMetropolis,
+    CompoundStep,
     HamiltonianMC,
     Metropolis,
     Slice,
 )
-from step_methods.compound import CompoundStep
 from tests.helpers import SeededTest, fast_unstable_sampling_mode
 from tests.models import simple_init