black

BAMresearch · Aug 30, 2024 · 20f8502 · 20f8502
1 parent 4d15278
commit 20f8502
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/probeye/inference/bias/solver.py b/probeye/inference/bias/solver.py
@@ -329,7 +329,9 @@ def restart_run(self, state, n_steps):
             Number of steps to run.
         """
 
-        self.sampler.run_mcmc(initial_state=state, nsteps=n_steps, progress=self.show_progress)
+        self.sampler.run_mcmc(
+            initial_state=state, nsteps=n_steps, progress=self.show_progress
+        )
         self.var_names = self.problem.get_theta_names(tex=True, components=True)
         inference_data = az.from_emcee(self.sampler, var_names=self.var_names)
-        return inference_data
+        return inference_data