diff --git a/.circleci/config.yml b/.circleci/config.yml index b4d27d2d6..a095d326c 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -10,7 +10,7 @@ jobs: parallelism: 1 steps: - checkout - - run: sudo pip install .[sklearn,tf,torch,testing] + - run: sudo pip install .[sklearn,tf-cpu,torch,testing] - run: sudo pip install codecov pytest-cov - run: python -m pytest -n 8 --dist=loadfile -s -v ./tests/ --cov - run: codecov @@ -26,7 +26,7 @@ jobs: parallelism: 1 steps: - checkout - - run: sudo pip install .[mecab,sklearn,tf,torch,testing] + - run: sudo pip install .[mecab,sklearn,tf-cpu,torch,testing] - run: python -m pytest -n 8 --dist=loadfile -s -v ./tests/ run_tests_torch: working_directory: ~/transformers @@ -52,7 +52,7 @@ jobs: parallelism: 1 steps: - checkout - - run: sudo pip install .[sklearn,tf,testing] + - run: sudo pip install .[sklearn,tf-cpu,testing] - run: sudo pip install codecov pytest-cov - run: python -m pytest -n 8 --dist=loadfile -s -v ./tests/ --cov - run: codecov diff --git a/setup.py b/setup.py index b36d51e71..81fb09f85 100644 --- a/setup.py +++ b/setup.py @@ -63,6 +63,7 @@ extras = {} extras["mecab"] = ["mecab-python3"] extras["sklearn"] = ["scikit-learn"] extras["tf"] = ["tensorflow"] +extras["tf-cpu"] = ["tensorflow-cpu"] extras["torch"] = ["torch"] extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"]