From 8b655f4a2d800bcda78359e0b46d837d9e55be91 Mon Sep 17 00:00:00 2001
From: Nikita Savelyev <nikita.savelyev@intel.com>
Date: Tue, 10 Dec 2024 15:25:23 +0100
Subject: [PATCH] Remove openvino-tests extra requirement

---
 .github/workflows/test_openvino.yml      | 2 +-
 .github/workflows/test_openvino_full.yml | 2 +-
 .github/workflows/test_openvino_slow.yml | 2 +-
 setup.py                                 | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
index 67f6d680a..7583c5107 100644
--- a/.github/workflows/test_openvino.yml
+++ b/.github/workflows/test_openvino.yml
@@ -43,7 +43,7 @@ jobs:
         run: |
           pip install --upgrade pip
           pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-          pip install .[openvino,openvino-tokenizers,diffusers,tests,tests-openvino] transformers[testing]
+          pip install .[openvino,openvino-tokenizers,diffusers,tests] transformers[testing]
 
       - if: ${{ matrix.transformers-version != 'latest' }}
         name: Downgrade Transformers and Accelerate
diff --git a/.github/workflows/test_openvino_full.yml b/.github/workflows/test_openvino_full.yml
index 3455f8ca5..914035b75 100644
--- a/.github/workflows/test_openvino_full.yml
+++ b/.github/workflows/test_openvino_full.yml
@@ -56,7 +56,7 @@ jobs:
           python -m pip install --upgrade pip
           # Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages
           pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-          pip install .[tests,tests-openvino]
+          pip install .[tests]
 
       - name: Install openvino-nightly
         if: ${{ matrix.openvino == 'ov-nightly' }}
diff --git a/.github/workflows/test_openvino_slow.yml b/.github/workflows/test_openvino_slow.yml
index f7555c64b..9ad5ef269 100644
--- a/.github/workflows/test_openvino_slow.yml
+++ b/.github/workflows/test_openvino_slow.yml
@@ -42,7 +42,7 @@ jobs:
         run: |
           pip install --upgrade pip
           pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-          pip install .[openvino,tests,tests-openvino] transformers[testing]
+          pip install .[openvino,tests] transformers[testing]
           pip uninstall -y nncf
 
       - if: ${{ matrix.transformers-version != 'latest' }}
diff --git a/setup.py b/setup.py
index 8c8c19f11..64b638925 100644
--- a/setup.py
+++ b/setup.py
@@ -56,6 +56,7 @@
     "sentence-transformers",
     "open_clip_torch>=2.26.1",
     "peft",
+    "datasets[audio]>=1.4.0",
 ]
 
 QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
@@ -68,7 +69,6 @@
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,
-    "tests-openvino": ["datasets[audio]>=1.4.0"],
 }
 
 setup(