From b6d7392579286b6dbd8ca96c0bcb4cc6f7c3c4a0 Mon Sep 17 00:00:00 2001
From: Roger Wang <136131678+ywang96@users.noreply.github.com>
Date: Sun, 29 Sep 2024 21:28:26 -0700
Subject: [PATCH] [Misc][CI/Build] Include `cv2` via `mistral_common[opencv]` 
 (#8951)

---
 requirements-common.txt | 2 +-
 setup.py                | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/requirements-common.txt b/requirements-common.txt
index 855169aa..aa165ff6 100644
--- a/requirements-common.txt
+++ b/requirements-common.txt
@@ -26,7 +26,7 @@ pyzmq
 msgspec
 gguf == 0.10.0
 importlib_metadata
-mistral_common >= 1.4.3
+mistral_common[opencv] >= 1.4.4
 pyyaml
 six>=1.16.0; python_version > '3.11' # transitive dependency of pandas that needs to be the latest version for python 3.12
 setuptools>=74.1.1; python_version > '3.11' # Setuptools is used by triton, we need to ensure a modern version is installed for 3.12+ so that it does not try to import distutils, which was removed in 3.12
diff --git a/setup.py b/setup.py
index 26ed33f8..759e1c5f 100644
--- a/setup.py
+++ b/setup.py
@@ -512,7 +512,6 @@ setup(
     ext_modules=ext_modules,
     extras_require={
         "tensorizer": ["tensorizer>=2.9.0"],
-        "video": ["opencv-python"],  # Required for video processing
         "audio": ["librosa", "soundfile"]  # Required for audio processing
     },
     cmdclass={"build_ext": cmake_build_ext} if len(ext_modules) > 0 else {},