From f8c47e3eec6303c6026c04d156cfcf6184d3c378 Mon Sep 17 00:00:00 2001
From: Jack Elvin-Poole <jack.elvinpoole@gmail.com>
Date: Fri, 15 Mar 2024 16:19:33 -0400
Subject: [PATCH] Update Dockerfile

jax version change
---
 txpipe-dev/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/txpipe-dev/Dockerfile b/txpipe-dev/Dockerfile
index ef8023b..d9b7c2c 100644
--- a/txpipe-dev/Dockerfile
+++ b/txpipe-dev/Dockerfile
@@ -49,8 +49,8 @@ RUN mamba install -c conda-forge -y scipy=1.8.0 \
                      pygraphviz \
                      compilers \
                      guppy3 \
-                     "jax=0.4.14" \
-                     "jaxlib=0.4.14=*cpu*" \
+                     "jax=0.4.19" \
+                     "jaxlib=0.4.19=*cpu*" \
                      "dm-tree=0.1.7" \
                      "cosmosis=3.4" \
                      threadpoolctl \