diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 2cb554a..f2547c4 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -16,7 +16,7 @@ jobs:
   checks:
     strategy:
       matrix:
-        python-version: ["3.9", "3.10"]
+        python-version: ["3.9", "3.10", 3.11", "3.12"]
     uses: ecmwf-actions/reusable-workflows/.github/workflows/qa-pytest-pyproject.yml@v2
     with:
       python-version: ${{ matrix.python-version }}
diff --git a/.github/workflows/python-pull-request.yml b/.github/workflows/python-pull-request.yml
index 2185264..c2be6a4 100644
--- a/.github/workflows/python-pull-request.yml
+++ b/.github/workflows/python-pull-request.yml
@@ -17,7 +17,7 @@ jobs:
   checks:
     strategy:
       matrix:
-        python-version: ["3.9", "3.10"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
     uses: ecmwf-actions/reusable-workflows/.github/workflows/qa-pytest-pyproject.yml@v2
     with:
       python-version: ${{ matrix.python-version }}
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 9bd2885..53dd832 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -22,6 +22,7 @@ Keep it human-readable, your future self will thank you!
 - Add missing classes in checkpoint handling
 - Rename Condition to State [#24](https://github.com/ecmwf/anemoi-inference/pull/24)
 - Fix pre-commit regex
+- ci: extend python versions to include 3.11 and 3.12 [#31] (https://github.com/ecmwf/anemoi-inference/pull/31)
 
 ### Removed
 
diff --git a/src/anemoi/inference/checkpoint/__init__.py b/src/anemoi/inference/checkpoint/__init__.py
index 5bec6c0..ebb28b7 100644
--- a/src/anemoi/inference/checkpoint/__init__.py
+++ b/src/anemoi/inference/checkpoint/__init__.py
@@ -78,8 +78,7 @@ def validate_environment(
         *,
         exempt_packages: list[str] | None = None,
     ) -> bool:
-        """
-        Validate environment of the checkpoint against the current environment.
+        """Validate environment of the checkpoint against the current environment.
 
         Parameters
         ----------
diff --git a/src/anemoi/inference/checkpoint/metadata/version_0_0_0.py b/src/anemoi/inference/checkpoint/metadata/version_0_0_0.py
index 83ab687..6faff1e 100644
--- a/src/anemoi/inference/checkpoint/metadata/version_0_0_0.py
+++ b/src/anemoi/inference/checkpoint/metadata/version_0_0_0.py
@@ -14,8 +14,7 @@
 
 
 class Version_0_0_0(Metadata):
-    """
-    Reference for very old checkpoints
+    """Reference for very old checkpoints
     Will not work and need to be updated
     """
 
diff --git a/src/anemoi/inference/checkpoint/metadata/version_0_1_0.py b/src/anemoi/inference/checkpoint/metadata/version_0_1_0.py
index bf10ab7..1367fba 100644
--- a/src/anemoi/inference/checkpoint/metadata/version_0_1_0.py
+++ b/src/anemoi/inference/checkpoint/metadata/version_0_1_0.py
@@ -23,9 +23,7 @@ def __init__(self, metadata):
 
     @cached_property
     def _dataset(self):
-        """
-        Part of the metadata that refers to the zarr dataset
-        """
+        """Part of the metadata that refers to the zarr dataset."""
         return self._metadata["dataset"]
 
     @cached_property
diff --git a/src/anemoi/inference/checkpoint/metadata/version_0_2_0.py b/src/anemoi/inference/checkpoint/metadata/version_0_2_0.py
index 2bb67ca..573c56c 100644
--- a/src/anemoi/inference/checkpoint/metadata/version_0_2_0.py
+++ b/src/anemoi/inference/checkpoint/metadata/version_0_2_0.py
@@ -16,8 +16,7 @@
 
 
 class DataRequest:
-    """
-    Base class for all data requests.
+    """Base class for all data requests.
 
     Data requests describe operations on the input data that are needed to prepare it for inference.
     The same operations that were applied to the training dataset should be applied to the input data.