Revert "nltk: fix broken punkt download" (livekit#630)

SuJingnan · Aug 15, 2024 · c7e0ab1 · c7e0ab1
1 parent 8f563a8
commit c7e0ab1
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/.changeset/funny-squids-help.md b/.changeset/funny-squids-help.md
@@ -0,0 +1,5 @@
+---
+"livekit-plugins-nltk": patch
+---
+
+Revert "nltk: fix broken punkt download"
diff --git a/.changeset/itchy-ligers-exist.md b/.changeset/itchy-ligers-exist.md
diff --git a/livekit-plugins/livekit-plugins-nltk/livekit/plugins/nltk/__init__.py b/livekit-plugins/livekit-plugins-nltk/livekit/plugins/nltk/__init__.py
@@ -32,9 +32,9 @@ def __init__(self):
 
     def download_files(self):
         try:
-            _ = nltk.data.find("tokenizers/punkt_tab")
+            _ = nltk.data.find("tokenizers/punkt")
         except LookupError:
-            nltk.download("punkt_tab")
+            nltk.download("punkt")
 
 
 Plugin.register_plugin(NltkPlugin())
diff --git a/livekit-plugins/livekit-plugins-nltk/setup.py b/livekit-plugins/livekit-plugins-nltk/setup.py
@@ -46,7 +46,7 @@
     license="Apache-2.0",
     packages=setuptools.find_namespace_packages(include=["livekit.*"]),
     python_requires=">=3.9.0",
-    install_requires=["livekit-agents>=0.8.0.dev0", "nltk >= 3.8.2, < 4"],
+    install_requires=["livekit-agents>=0.8.0.dev0", "nltk >= 3, < 4"],
     package_data={"livekit.plugins.nltk": ["py.typed"]},
     project_urls={
         "Documentation": "https://docs.livekit.io",