From bd382c1ce8020b142fb4d849565245746d254ea4 Mon Sep 17 00:00:00 2001
From: Luke Baumann <lukebaumann@google.com>
Date: Thu, 26 Feb 2026 19:37:19 +0000
Subject: [PATCH] Updating the requirements.txt file and adding packaging as a
 dependency

---
 pyproject.toml   | 1 +
 requirements.txt | 7 ++++---
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 49c4d06..4590269 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,6 +18,7 @@ dependencies = [
     "orbax-checkpoint",
     "uvicorn",
     "requests",
+    "packaging",
 ]
 
 # `version` is automatically set by flit to use `my_project.__version__`
diff --git a/requirements.txt b/requirements.txt
index 1d0cd4a..0320364 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
-jax[cpu]>=0.5.1
 absl-py
+fastapi
+jax>=0.7.2
 orbax-checkpoint
 uvicorn
-fastapi
-google-cloud-logging
+requests
+packaging