Add pre-commit hooks for formatting and nb stripping

.pre-commit-config.yaml

@@ -1,5 +1,5 @@
 repos:
-  # General linting:
+  # General linting
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v3.2.0
     hooks:
@@ -7,3 +7,14 @@ repos:
       - id: end-of-file-fixer
       - id: check-yaml
       - id: check-added-large-files
+  # General formatting
+  - repo: https://github.com/psf/black
+    rev: 23.3.0
+    hooks:
+      - id: black
+      - id: black-jupyter
+  # Stripping notebooks
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.6.1
+    hooks:
+      - id: nbstripout

benchmarks/print_best_model.py

@@ -51,7 +51,6 @@ import glob
 path = TRIALS_FOLDER + "/*.pkl"
 files = 0
 for fname in glob.glob(path):
-
     trials_obj = pkl.load(open(fname, "rb"))
     n_trials = trials_obj["n"]
     trials_obj = trials_obj["trials"]