From 4d222f5217a6608c6edbb9f23ef75462dfbe29e9 Mon Sep 17 00:00:00 2001
From: markus583 <markus.frohmann@gmail.com>
Date: Sun, 16 Jun 2024 13:36:15 +0000
Subject: [PATCH] fix reqs

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 1e72bade..43ec95ee 100644
--- a/setup.py
+++ b/setup.py
@@ -10,13 +10,14 @@
     install_requires=[
         "onnxruntime>=1.13.1",
         "transformers>=4.22.2",
-        "numpy>=1",
+        "numpy>=1.0,<=2.0",
         "scikit-learn>=1",
         "tqdm",
         "skops",
         "pandas>=1",
         "cached_property",  # for Py37
         "torchinfo",
+        "fast-mosestokenizer"
     ],
     url="https://github.com/bminixhofer/wtpsplit",
     package_data={"wtpsplit": ["data/*"]},