tests reqs

openvinotoolkit · Feb 6, 2024 · 1be57bb · 1be57bb
1 parent 60fbf91
commit 1be57bb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.github/workflows/linux.yml b/.github/workflows/linux.yml
@@ -281,7 +281,7 @@ jobs:
           # Find and install wheel
           pushd ${INSTALL_DIR}/ov_tokenizers
             wheel_name=$(find . -name 'openvino_tokenizers*.whl')
-            python3 -m pip install $wheel_name[all]
+            python3 -m pip install $wheel_name[test]
           popd
 
       - name: Tokenizers Bandit tests

diff --git a/pyproject.toml b/pyproject.toml
@@ -28,11 +28,21 @@ transformers = [
 ]
 dev = [
     "ruff",
+    "bandit",
     "pytest",
     "pytest_harvest",
-    "bandit",
     "openvino_tokenizers[transformers]"
 ]
+torch = [
+    'torch @ https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.0.1%2Bcpu.cxx11.abi-cp38-cp38-linux_x86_64.whl ; sys_platform=="linux" and python_version == "3.8"',
+    'torch @ https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.0.1%2Bcpu.cxx11.abi-cp39-cp39-linux_x86_64.whl ; sys_platform=="linux" and python_version == "3.9"',
+    'torch @ https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.0.1%2Bcpu.cxx11.abi-cp310-cp310-linux_x86_64.whl ; sys_platform=="linux" and python_version == "3.10"',
+    'torch @ https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.0.1%2Bcpu.cxx11.abi-cp311-cp311-linux_x86_64.whl ; sys_platform=="linux" and python_version == "3.11"',
+    'torch ; sys_platform != "linux"'
+]
+test = [
+    "openvino_tokenizers[dev, torch]"
+]
 fuzzing = [
     "atheris",
     "openvino_tokenizers[transformers]"