From 11a081e09e92771e51a5d2758d53a9afb59547f0 Mon Sep 17 00:00:00 2001 From: elisim Date: Tue, 7 Feb 2023 18:46:22 +0700 Subject: [PATCH] remove unused libs for this PR for creating the env --- setup.py | 43 +++++++++++++++++++++---------------------- 1 file changed, 21 insertions(+), 22 deletions(-) diff --git a/setup.py b/setup.py index aad145b145f..603760581fa 100644 --- a/setup.py +++ b/setup.py @@ -123,7 +123,6 @@ _deps = [ "jax>=0.2.8,!=0.3.2,<=0.3.6", "jaxlib>=0.1.65,<=0.3.6", "jieba", - "kenlm", "keras-nlp>=0.3.1", "nltk", "natten>=0.14.4", @@ -159,9 +158,9 @@ _deps = [ "sigopt", "librosa", "starlette", - "tensorflow-cpu>=2.4,<2.12", - "tensorflow>=2.4,<2.12", - "tensorflow-text", + # "tensorflow-cpu>=2.4,<2.12", + # "tensorflow>=2.4,<2.12", + # "tensorflow-text", "tf2onnx", "timeout-decorator", "timm", @@ -176,7 +175,7 @@ _deps = [ "uvicorn", "beautifulsoup4", "sudachipy>=0.6.6", - "sudachidict_core>=20220729", + # "sudachidict_core>=20220729", "rhoknp>=1.1.0", ] @@ -246,11 +245,11 @@ class DepsTableUpdateCommand(Command): extras = {} -extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic", "sudachipy", "sudachidict_core", "rhoknp") +extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic", "sudachipy", "rhoknp") extras["sklearn"] = deps_list("scikit-learn") -extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "tf2onnx", "tensorflow-text", "keras-nlp") -extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "tf2onnx", "tensorflow-text", "keras-nlp") +extras["tf"] = deps_list("tf2onnx") +extras["tf-cpu"] = deps_list("tf2onnx") extras["torch"] = deps_list("torch") extras["accelerate"] = deps_list("accelerate") @@ -278,7 +277,7 @@ extras["sigopt"] = deps_list("sigopt") extras["integrations"] = extras["optuna"] + extras["ray"] + extras["sigopt"] extras["serving"] = deps_list("pydantic", "uvicorn", "fastapi", "starlette") -extras["audio"] = deps_list("librosa", "pyctcdecode", "phonemizer", "kenlm") +extras["audio"] = deps_list("librosa", "pyctcdecode", "phonemizer") # `pip install ".[speech]"` is deprecated and `pip install ".[torch-speech]"` should be used instead extras["speech"] = deps_list("torchaudio") + extras["audio"] extras["torch-speech"] = deps_list("torchaudio") + extras["audio"] @@ -363,19 +362,19 @@ extras["dev-torch"] = ( + extras["modelcreation"] + extras["onnxruntime"] ) -extras["dev-tensorflow"] = ( - extras["testing"] - + extras["tf"] - + extras["sentencepiece"] - + extras["tokenizers"] - + extras["vision"] - + extras["quality"] - + extras["docs_specific"] - + extras["sklearn"] - + extras["modelcreation"] - + extras["onnx"] - + extras["tf-speech"] -) +# extras["dev-tensorflow"] = ( +# extras["testing"] +# + extras["tf"] +# + extras["sentencepiece"] +# + extras["tokenizers"] +# + extras["vision"] +# + extras["quality"] +# + extras["docs_specific"] +# + extras["sklearn"] +# + extras["modelcreation"] +# + extras["onnx"] +# + extras["tf-speech"] +# ) extras["dev"] = ( extras["all"] + extras["testing"]