Pin to numpy<2 to avoid spacy compat issues (NVIDIA#119)

Signed-off-by: Ayush Dattagupta <[email protected]>
sarahyurick · Jun 18, 2024 · 9da54b8 · 9da54b8
1 parent f1e993b
commit 9da54b8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -64,6 +64,9 @@
         # justext installation breaks without lxml[html_clean]
         # due to this: https://github.com/miso-belica/jusText/issues/47
         "lxml[html_clean]",
+        # Numpy 2.0 breaks with spacy https://github.com/explosion/spaCy/issues/13528
+        # TODO: Remove when issue is fixed
+        "numpy<2",
     ],
     extras_require={
         "cuda12x": [