From a17c6e392e4d076dcb0100f6f57fbe1e7e1aeac1 Mon Sep 17 00:00:00 2001
From: Li Jiang <bnujli@gmail.com>
Date: Mon, 12 Aug 2024 08:14:21 +0800
Subject: [PATCH] Fix test errors of nltk and numpy (#1335)

* Fix test errors with nltk 3.8.2

* Fix test errors with numpy large

* Fix test errors with numpy large
---
 setup.py                     | 6 +++---
 test/automl/test_forecast.py | 4 ++++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 14adb5ba7..73a1b6fb0 100644
--- a/setup.py
+++ b/setup.py
@@ -73,7 +73,7 @@ setuptools.setup(
             "dataclasses",
             "transformers[torch]==4.26",
             "datasets",
-            "nltk",
+            "nltk<=3.8.1",  # 3.8.2 doesn't work with mlflow
             "rouge_score",
             "hcrystalball==0.1.10",
             "seqeval",
@@ -117,14 +117,14 @@ setuptools.setup(
         "hf": [
             "transformers[torch]==4.26",
             "datasets",
-            "nltk",
+            "nltk<=3.8.1",
             "rouge_score",
             "seqeval",
         ],
         "nlp": [  # for backward compatibility; hf is the new option name
             "transformers[torch]==4.26",
             "datasets",
-            "nltk",
+            "nltk<=3.8.1",
             "rouge_score",
             "seqeval",
         ],
diff --git a/test/automl/test_forecast.py b/test/automl/test_forecast.py
index c305e78ae..8f0a24a1c 100644
--- a/test/automl/test_forecast.py
+++ b/test/automl/test_forecast.py
@@ -151,6 +151,10 @@ def test_numpy():
     print(automl.predict(12))
 
 
+@pytest.mark.skipif(
+    sys.platform in ["darwin"],
+    reason="do not run on mac os",
+)
 def test_numpy_large():
     import numpy as np
     import pandas as pd