Develop

.flake8

@@ -1,6 +1,6 @@
 [flake8]
 max-line-length = 130
-ignore = W504, W503, E266
+ignore = W504, W503, E266, D, BLK
 exclude =
     .git,
     __pycache__,

.gitignore

@@ -7,6 +7,7 @@ __pycache__
 python/scratch
 .idea/*
 .vscode/
+*.code-workspace
 *checkpoint.ipynb
 build/
 venv/

brainbox/behavior/training.py

@@ -52,7 +52,6 @@
 from one.api import ONE
 from one.alf.io import AlfBunch
 from one.alf.exceptions import ALFObjectNotFound
-
 import psychofit as psy
 
 _logger = logging.getLogger('ibllib')
@@ -83,15 +82,16 @@ class TrainingStatus(IntFlag):
     ... assert TrainingStatus[status.upper()] in ~TrainingStatus.FAILED, 'Subject untrained'
     ... assert TrainingStatus[status.upper()] in TrainingStatus.TRAINED ^ TrainingStatus.READY
 
-    # Get the next training status
+    Get the next training status
+
     >>> next(member for member in sorted(TrainingStatus) if member > TrainingStatus[status.upper()])
     <TrainingStatus.READY4RECORDING: 128>
 
     Notes
     -----
     - ~TrainingStatus.TRAINED means any status but trained 1a or trained 1b.
     - A subject may acheive both TRAINED_1A and TRAINED_1B within a single session, therefore it
-     is possible to have skipped the TRAINED_1A session status.
+      is possible to have skipped the TRAINED_1A session status.
     """
     UNTRAINABLE = auto()
     UNBIASABLE = auto()