Add [math] optional dependency, make antlr req more stringent

2906a565 · Hailey Schoelkopf · GitHub · c8f56b17 · 2906a565
Unverified Commit 2906a565 authored Sep 21, 2023 by Hailey Schoelkopf Committed by GitHub Sep 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

pyproject.toml pyproject.toml +1 -2

No files found.
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,8 +36,6 @@ dependencies = [
    "tqdm-multiprocess",
    "transformers>=4.1",
    "zstandard",
-    "sympy>=1.12",
-    "antlr4-python3-runtime >= 4.11, < 4.12"
 ]

 [tool.setuptools]
@@ -66,6 +64,7 @@ linting = [
 ]
 testing = ["pytest", "pytest-cov", "pytest-xdist"]
 multilingual = ["nagisa>=0.2.7", "jieba>=0.42.1", "pycountry"]
+math = ["sympy>=1.12", "antlr4-python3-runtime==4.11"]
 sentencepiece = ["sentencepiece>=0.1.98", "protobuf>=4.22.1"]
 promptsource = [
    "promptsource @ git+https://github.com/bigscience-workshop/promptsource.git#egg=promptsource"