pymc-devs · AlexAndorra · Oct 30, 2020 · Oct 29, 2020 · Oct 29, 2020 · Oct 30, 2020
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -22,10 +22,12 @@ repos:
 - repo: local
   hooks:
     - id: watermark
-      name: Check notebooks have watermark
+      name: Check notebooks have watermark (see Jupyter style guide from PyMC3 Uncyclo)
       types: [jupyter]
-      entry: python scripts/check_watermark.py
-      language: python
+      entry: '%load_ext watermark.*%watermark -n -u -v -iv -w'
+      language: pygrep
+      args: [--negate, --multiline]
+      minimum_pre_commit_version: 2.8.0
     - id: check-toc
       name: Check all notebooks appear in table of contents
       types: [jupyter]

diff --git a/environment-dev.yml b/environment-dev.yml
@@ -21,6 +21,7 @@ dependencies:
   - nose>=1.3
   - nose-parameterized>=0.6
   - numpydoc>=0.9
+  - pre-commit>=2.8.0
   - pycodestyle>=2.3
   - pyflakes>=1.5
   - pylint>=1.7

diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -6,6 +6,7 @@ nbsphinx>=0.4.2
 nose>=1.3.7
 nose-parameterized==0.6.0
 numpydoc>=0.9.1
+pre-commit>=2.8.0
 pycodestyle>=2.3.1
 pyflakes>=1.5.0
 pylint>=1.7.4

diff --git a/scripts/check_watermark.py b/scripts/check_watermark.py