From 4b71941f590948bdc08f684b8cbed7b88baf87c4 Mon Sep 17 00:00:00 2001
From: stoprightthere <pmostowsky@gmail.com>
Date: Sat, 21 Feb 2026 21:14:34 +0300
Subject: [PATCH] Pin orbax-checkpoint version for Windows

Compatibility issue with a newer gpjax:
```
`uvloop` (v0.22.1) was included because `geometric-kernels:dev`
        (v0.4.2) depends on `gpjax` (v0.13.6) which depends on `flax`
        (v0.12.3) which depends on `orbax-checkpoint` (v0.11.33) which depends
        on `uvloop`
```
but
```
RuntimeError: uvloop does not support Windows at the moment
```
---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index a8432136..cd5db34f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -105,5 +105,6 @@ dev = [
   'jaxlib',
   'jaxtyping',
   'optax',
-  'gpjax>=0.12.2; python_version >= "3.10"',  # gpjax requires python >= 3.10.
+  'gpjax>=0.12.2',
+  'orbax-checkpoint==0.11.32; platform_system == "Windows"',   # newer version depends on uvloop, unavailable in windows
 ]