From 01ee9443d620a9c26f1eaa262a013bdbec8a5045 Mon Sep 17 00:00:00 2001
From: Sam-Armstrong <samuel_e_armstrong@yahoo.co.uk>
Date: Wed, 18 Sep 2024 13:43:30 +0100
Subject: [PATCH] fix: pin framework versions in optional.txt

---
 requirements/optional.txt | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/requirements/optional.txt b/requirements/optional.txt
index bad7809c6c56..07d8207532d1 100644
--- a/requirements/optional.txt
+++ b/requirements/optional.txt
@@ -5,9 +5,9 @@ coverage
 dm-haiku # mod_name=haiku
 flax
 h5py
-hypothesis==6.98.10 # fixed version till this get solved: https://github.com/HypothesisWorks/hypothesis/issues/3896
-jax
-jaxlib
+hypothesis==6.98.10
+jax<=0.4.33
+jaxlib<=0.4.33
 matplotlib
 ml-dtypes # mod_name=ml_dtypes
 networkx
@@ -24,7 +24,7 @@ redis
 scikit-learn # mod_name=sklearn
 scipy
 snakeviz # for profiling
-tensorflow
-torch
-torchvision
+tensorflow<=2.17.0
+torch<=2.4.1
+torchvision<=0.19.1
 xgboost