Bump python version (#645)

* Bump python version

* update bool inverse

.github/workflows/nightly.yml

@@ -14,7 +14,7 @@ jobs:
         fetch-depth: 0
     - uses: actions/setup-python@v4
       with:
-        python-version: 3.9
+        python-version: 3.11
     - name: Update pyproject.toml
       # Taken from https://github.com/aesara-devs/aesara/pull/1375
       run: |

.github/workflows/publish_documentation.yml

@@ -14,10 +14,10 @@ jobs:
         with:
           persist-credentials: false
 
-      - name: Set up Python 3.9
+      - name: Set up Python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.9
+          python-version: 3.11
 
       - name: Build the documentation with Sphinx
         run: |

.github/workflows/release.yml

@@ -14,7 +14,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v4
       with:
-        python-version: 3.9
+        python-version: 3.11
     - name: Build sdist and wheel
       run: |
         python -m pip install -U pip
@@ -51,7 +51,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v4
       with:
-        python-version: 3.9
+        python-version: 3.11
     - name: Give PyPI some time to update the index
       run: sleep 240
     - name: Attempt install from PyPI

.github/workflows/test.yml

@@ -14,7 +14,7 @@ jobs:
     - uses: actions/checkout@v3
     - uses: actions/setup-python@v4
       with:
-        python-version: 3.9
+        python-version: 3.11
     - uses: pre-commit/[email protected]
 
   test:
@@ -24,7 +24,7 @@ jobs:
       - style
     strategy:
       matrix:
-        python-version: [ '3.9', '3.11']
+        python-version: ['3.11', '3.12']
     steps:
     - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}

blackjax/mcmc/trajectory.py

@@ -385,7 +385,7 @@ def buildtree_integrate(
                 initial_energy,
             )
             # Note that is_diverging and is_turning is inplace updated
-            if ~is_diverging & ~is_turning:
+            if (not is_diverging) & (not is_turning):
                 start_state = jax.lax.cond(
                     direction > 0,
                     lambda _: trajectory.rightmost_state,
@@ -411,7 +411,7 @@ def buildtree_integrate(
                 )
                 trajectory = merge_trajectories(left_trajectory, right_trajectory)
 
-                if ~is_turning:
+                if not is_turning:
                     is_turning = uturn_check_fn(
                         trajectory.leftmost_state.momentum,
                         trajectory.rightmost_state.momentum,