From 1026f5f9add898c3a958c02c63ad7a8b7cf24c86 Mon Sep 17 00:00:00 2001 From: James Thewlis Date: Thu, 15 Aug 2024 16:41:08 +0100 Subject: [PATCH] Add dev dependency for numpy<2 The pytorch lightning version we're using for training is incompatible with numpy 2. We should update the training code to work with newer lightning, but in the meantime set an upper bound for numpy in the dev deps. --- pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/pyproject.toml b/pyproject.toml index 816e6e2..4887b69 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,6 +33,7 @@ dev = [ "scikit-learn >= 0.23.2", "tqdm", "pre-commit", + "numpy<2" ] [tool.black]