diff --git a/requirements.txt b/requirements.txt index 6fc8cd05..2bb52f68 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ +numpy<2.0 six pandas>=0.20.3 tensorflow>=2.15.0 diff --git a/setup.py b/setup.py index b4027727..4d4e3de2 100644 --- a/setup.py +++ b/setup.py @@ -39,6 +39,7 @@ "appdirs", "scikit-learn", "mhcgnomes>=0.8.4", + "numpy<2.0", "pyyaml", "tqdm", "tensorflow>=2.15.0",