diff --git a/models/ModelBase.py b/models/ModelBase.py index 217147d..b8b38d3 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -297,6 +297,7 @@ class ModelBase(object): def ask_target_iter(self, default_value=0): print("Current directory") cmd = "pwd" + print("Dentro ask_target_iter: \n\n") os.system(cmd) answer_filename = 'workspace/interact/interact_dict.pkl' if os.path.exists(answer_filename): @@ -307,6 +308,9 @@ class ModelBase(object): print("\n\n\n\nDefault target iteration: \n\n\n", default_target_iter) self.options['target_iter'] = max(0, default_target_iter + 10000) print("TARGET ITERATION: " + str(self.options['target_iter'])) + else: + default_target_iter = self.load_or_def_option('target_iter', default_value) + self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter)) else: default_target_iter = self.load_or_def_option('target_iter', default_value) self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))