Set up nbqa pre-commit hook

.pre-commit-config.yaml

@@ -1,29 +1,38 @@
 # See https://pre-commit.com for more information
 # See https://pre-commit.com/hooks.html for more hooks
 repos:
--   repo: https://gitlab.com/pycqa/flake8
-    rev: master
-    hooks:
-    -   id: flake8
-        args: ['src']
--   repo: https://github.com/Yelp/detect-secrets
-    rev: v0.13.1
-    hooks:
-    -   id: detect-secrets
-        args: ['--baseline', '.secrets.baseline']
-        exclude: .*/tests/.*
--   repo: https://github.com/aflc/pre-commit-jupyter
-    rev: v1.1.0
-    hooks:
-    -   id: jupyter-notebook-cleanup
-        args:
-        -   --remove-kernel-metadata
-        -   --pin-patterns
-        -   "[keep_output]"
--   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v2.4.0
-    hooks:
-    -   id: check-added-large-files
-        args: ['--maxkb=5120']
-    -   id: end-of-file-fixer
-    -   id: trailing-whitespace
+    - repo: https://github.com/psf/black
+      rev: 19.10b0 # Replace by any tag/version: https://github.com/psf/black/tags
+      hooks:
+          - id: black
+            language_version: python3 # Should be a command that runs python3.6+
+    - repo: https://github.com/Yelp/detect-secrets
+      rev: v0.13.1
+      hooks:
+          - id: detect-secrets
+            args: ["--baseline", ".secrets.baseline"]
+            exclude: .*/tests/.*
+    - repo: https://github.com/aflc/pre-commit-jupyter
+      rev: v1.1.0
+      hooks:
+          - id: jupyter-notebook-cleanup
+            args:
+                - --remove-kernel-metadata
+                - --pin-patterns
+                - "[keep_output]"
+    - repo: https://github.com/pre-commit/pre-commit-hooks
+      rev: v2.4.0
+      hooks:
+          - id: check-added-large-files
+            args: ["--maxkb=5120"]
+          - id: end-of-file-fixer
+          - id: trailing-whitespace
+    - repo: https://github.com/nbQA-dev/nbQA
+      rev: 0.5.4
+      hooks:
+          - id: nbqa-black
+            additional_dependencies: [black==20.8b1]
+          - id: nbqa-pyupgrade
+            additional_dependencies: [pyupgrade==2.7.3]
+          - id: nbqa-isort
+            additional_dependencies: [isort==5.6.4]

pyproject.toml

@@ -0,0 +1,7 @@
+[tool.nbqa.mutate]
+isort = 1
+black = 1
+pyupgrade = 1
+
+[tool.nbqa.addopts]
+pyupgrade = ["--py36-plus"]