ci: add autowalrus

zfit · Jan 9, 2024 · 66a589d · 66a589d
1 parent d976606
commit 66a589d
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 12 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -42,7 +42,7 @@ repos:
       - id: rst-directive-colons
 
   - repo: https://github.com/PyCQA/flake8
-    rev: 6.1.0
+    rev: 7.0.0
     hooks:
       - id: flake8
 
@@ -69,17 +69,22 @@ repos:
       - id: upgrade-type-hints
         args: ["--futures=true"]
 
+  -   repo: https://github.com/MarcoGorelli/auto-walrus
+      rev: v0.2.2
+      hooks:
+      -   id: auto-walrus
+
   - repo: https://github.com/psf/black
     rev: 23.12.1
     hooks:
       - id: black
 
-  - repo: https://github.com/pre-commit/mirrors-prettier
-    rev: v4.0.0-alpha.8
-    hooks:
-      - id: prettier
+#  - repo: https://github.com/pre-commit/mirrors-prettier
+#    rev: "v4.0.0-alpha.8"
+#    hooks:
+#      - id: prettier
 
-  - repo: https://github.com/roy-ht/pre-commit-jupyter
-    rev: v1.2.1
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.6.1
     hooks:
-      - id: jupyter-notebook-cleanup
+      - id: nbstripout
diff --git a/docs/GenMultiDecay_Tutorial.ipynb b/docs/GenMultiDecay_Tutorial.ipynb
@@ -3,7 +3,6 @@
   {
    "cell_type": "markdown",
    "metadata": {
-    "collapsed": true,
     "pycharm": {
      "is_executing": true,
      "name": "#%% md\n"
@@ -417,4 +416,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}
diff --git a/phasespace/phasespace.py b/phasespace/phasespace.py
@@ -200,8 +200,7 @@ def set_children(self, *children):
                 f"Have to set at least 2 children, not {len(children)} for a particle to decay"
             )
         # Check name clashes
-        name_clash = self._do_names_clash(children)
-        if name_clash:
+        if name_clash := self._do_names_clash(children):
             raise KeyError(f"Particle name {name_clash} already used")
         self.children = children
         return self

diff --git a/setup.cfg b/setup.cfg
@@ -23,6 +23,7 @@ classifiers =
     Programming Language :: Python :: 3.9
     Programming Language :: Python :: 3.10
     Programming Language :: Python :: 3.11
+    Programming Language :: Python :: 3.12
     Topic :: Scientific/Engineering :: Physics
 keywords = TensorFlow, phasespace, HEP