Fix training phase dialog

int-brain-lab · Sep 7, 2023 · 804cffc · 804cffc
1 parent d45c697
commit 804cffc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/iblrig/base_choice_world.py b/iblrig/base_choice_world.py
@@ -685,10 +685,12 @@ def __init__(self, training_phase=-1, **kwargs):
             try:
                 training_phase = get_training_phase(self.session_info.SUBJECT_NAME)
                 self.logger.warning(f"Got training phase: {training_phase}")
-            except Exception:
+            except Exception as ex:
+                self.logger.debug('Failed to get training phase: %s', ex)
                 if self.interactive:
                     training_phase = iblrig.graphic.numinput(
-                        "Subject training phase", "Subject trainig phase : (0-5)", nullable=False)
+                        "Subject training phase", "Subject training phase: (0-5)",
+                        askint=True, nullable=False, default=0, minval=0, maxval=5)
                 else:
                     self.logger.warning(f"Could not get training phase from Alyx: {traceback.format_exc()}, please set it"
                                         f"manually in ./iblrig_tasks/_iblrig_tasks_trainingChoiceWorld/task.py"