From 02abcbe2a2db1e6dafc9fe3922e0339b4741d11a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= Date: Mon, 16 May 2022 13:57:50 +0200 Subject: [PATCH] default pbar in SequentialRK4IP --- src/scgenerator/physics/simulate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/scgenerator/physics/simulate.py b/src/scgenerator/physics/simulate.py index 7cda137..3966990 100644 --- a/src/scgenerator/physics/simulate.py +++ b/src/scgenerator/physics/simulate.py @@ -134,7 +134,7 @@ class RK4IP: for num, state in self.irun(): if self.save_data: self._save_current_spectrum(state.actual_spectrum, num) - self.step_saved(state) + self.step_saved(state) self.logger.info( "propagation finished in {} steps ({} seconds)".format( @@ -224,10 +224,10 @@ class SequentialRK4IP(RK4IP): def __init__( self, params: Parameters, - pbars: PBars, + pbars: PBars = None, save_data=False, ): - self.pbars = pbars + self.pbars = pbars or PBars(params.z_num, f"Simulating {params.name}", 1) super().__init__( params, save_data=save_data,