Skip to content
Snippets Groups Projects
Commit 6e4da956 authored by Martin Millon's avatar Martin Millon
Browse files

fixing pipeline

parent d0eece7e
No related branches found
No related tags found
No related merge requests found
Pipeline #1103244444 passed
......@@ -151,8 +151,8 @@ class Optimizer(InferenceBase):
return params, opt_state, loss_val
# Gradient descent loop
for i in self._for_loop(range(max_iterations), progress_bar,
total=max_iterations,
for i in self._for_loop(range(int(max_iterations)), progress_bar,
total=int(max_iterations),
desc=f"optax.{self.method}"):
params, opt_state, loss_val = gradient_step(params, opt_state)
if stop_at_loss_increase and i > min_iterations and loss_val > prev_loss_val: #pragma: no cover
......
......@@ -105,9 +105,6 @@ class TestOptim(unittest.TestCase):
view_deconv_model(model, kwargs_final, data, sigma_2)
make_movie(model, kwargs_final, data, sigma_2, self.outpath, format='gif')
with self.assertRaises(RuntimeError):
make_movie(model, kwargs_final, data, sigma_2, self.outpath, format='xxx')
# This requires installation that are too heavy for the Docker. Skip this part
# make_movie(model, kwargs_final, data, sigma_2, self.outpath, format='mp4v')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment