diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
index 9c0895cad..a282ee1c8 100644
--- a/.github/workflows/python.yml
+++ b/.github/workflows/python.yml
@@ -32,6 +32,7 @@ jobs:
         # Re-install pytorch for cpu to avoid errors that cuda was not found
         run: |
           ./scripts/setup_py.sh
+          poetry run pip3 install --force-reinstall --pre torch==2.* --index-url https://download.pytorch.org/whl/nightly/cpu
 
       # We run testing before linting since testing, surprisingly, is much faster.
       - name: Test python
diff --git a/poetry.lock b/poetry.lock
index 23a018b2f..0557cd106 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1213,19 +1213,18 @@ tests = ["asttokens", "littleutils", "pytest", "rich"]
 
 [[package]]
 name = "fastapi"
-version = "0.99.1"
+version = "0.98.0"
 description = "FastAPI framework, high performance, easy to learn, fast to code, ready for production"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "fastapi-0.99.1-py3-none-any.whl", hash = "sha256:976df7bab51ac7beda9f68c4513b8c4490b5c1135c72aafd0a5ee4023ec5282e"},
-    {file = "fastapi-0.99.1.tar.gz", hash = "sha256:ac78f717cd80d657bd183f94d33b9bda84aa376a46a9dab513586b8eef1dc6fc"},
+    {file = "fastapi-0.98.0-py3-none-any.whl", hash = "sha256:f4165fb1fe3610c52cb1b8282c1480de9c34bc270f56a965aa93a884c350d605"},
+    {file = "fastapi-0.98.0.tar.gz", hash = "sha256:0d3c18886f652038262b5898fec6b09f4ca92ee23e9d9b1d1d24e429f84bf27b"},
 ]
 
 [package.dependencies]
 pydantic = ">=1.7.4,<1.8 || >1.8,<1.8.1 || >1.8.1,<2.0.0"
 starlette = ">=0.27.0,<0.28.0"
-typing-extensions = ">=4.5.0"
 
 [package.extras]
 all = ["email-validator (>=1.1.1)", "httpx (>=0.23.0)", "itsdangerous (>=1.1.0)", "jinja2 (>=2.11.2)", "orjson (>=3.2.1)", "python-multipart (>=0.0.5)", "pyyaml (>=5.3.1)", "ujson (>=4.0.1,!=4.0.2,!=4.1.0,!=4.2.0,!=4.3.0,!=5.0.0,!=5.1.0)", "uvicorn[standard] (>=0.12.0)"]
@@ -6074,4 +6073,4 @@ testing = ["big-O", "flake8 (<5)", "jaraco.functools", "jaraco.itertools", "more
 [metadata]
 lock-version = "2.0"
 python-versions = "~3.9"
-content-hash = "b30666a00cdc825e62804031b196ff2cbfecc898dc6b4b7067205d9f660b7a3f"
+content-hash = "f98e647131196995aef9095ee2bc1680ec3c3608f5681e648c53916f0ad59e80"
diff --git a/pyproject.toml b/pyproject.toml
index b57c53cfb..83e4eb5bc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,7 +15,7 @@ openai-function-call = "^0.0.5"
 
 datasets = "^2.12.0"
 duckdb = "^0.8.0"
-fastapi = "^0.99.1"
+fastapi = "^0.98.0"
 google-cloud-storage = "^2.5.0"
 python = "~3.9"
 pyarrow = "^9.0.0"