Skip internvideo2 Test if not installed (#1089)

* Skip internvideo2 test if not installed.

Signed-off-by: Ao Tang <aot@nvidia.com>

* skip at file level

Signed-off-by: Ao Tang <aot@nvidia.com>

* Add the installation script

Signed-off-by: Ao Tang <aot@nvidia.com>

* Add copyright header to installation script and update .gitignore to include InternVideo2 dependency

Signed-off-by: Ao Tang <aot@nvidia.com>

---------

Signed-off-by: Ao Tang <aot@nvidia.com>
Co-authored-by: Dong Hyuk Chang <thomaschang26@tutanota.com>

Author: suiyoubi

.gitignore

@@ -152,3 +152,6 @@ data/
 
 # Text Editor / IDE Files
 .vscode
+
+# InternVideo2 dependency (cloned by installation script)
+InternVideo/

external/intern_video2_installation.sh

@@ -0,0 +1,37 @@
+#!/bin/bash
+# Copyright (c) 2025, NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# =============================================================================
+# InternVideo2 Installation Script
+# =============================================================================
+# This script installs the InternVideo2 dependency for the Curator project.
+# It clones the official InternVideo repository, applies necessary patches,
+# and integrates it into the project's dependency management system using uv.
+
+# Verify that the script is being run from the correct directory
+# This ensures that relative paths in the script work correctly
+if [ "$(basename "$PWD")" != "Curator" ]; then
+  echo "Please run this script from the Curator/ directory."
+  exit 1
+fi
+
+# Clone the official InternVideo repository from OpenGVLab
+# This is the source repository for the InternVideo2 model implementation
+git clone https://github.com/OpenGVLab/InternVideo.git;
+cd InternVideo; git checkout 09d872e5093296c6f36b8b3a91fc511b76433bf7;
+
+# Apply a custom patch to the InternVideo2 codebase
+# This patch contains modifications needed for integration with NeMo Curator
+patch -p1 < ../external/intern_video2_multimodal.patch; cd ../

tests/models/test_internvideo2.py

@@ -18,12 +18,17 @@
 from unittest.mock import MagicMock, Mock, patch
 
 import numpy as np
+import pytest
 import torch
 
-from nemo_curator.stages.video.embedding.internvideo2 import (
-    InternVideo2EmbeddingStage,
-    InternVideo2FrameCreationStage,
-)
+try:
+    from nemo_curator.stages.video.embedding.internvideo2 import (
+        InternVideo2EmbeddingStage,
+        InternVideo2FrameCreationStage,
+    )
+except ImportError:
+    pytest.skip("InternVideo2 package not available", allow_module_level=True)
+
 from nemo_curator.tasks.video import Clip, Video, VideoTask
 
 # Create a random generator for consistent testing

tests/models/test_internvideo2_mm.py

@@ -21,14 +21,17 @@
 import torch
 from easydict import EasyDict
 
-from nemo_curator.models.internvideo2_mm import (
-    BERT_MODEL_ID,
-    INTERNVIDEO2_MODEL_FILE,
-    INTERNVIDEO2_MODEL_ID,
-    InternVideo2MultiModality,
-    _InternVideo2Stage2Wrapper,
-    _setup_internvideo2,
-)
+try:
+    from nemo_curator.models.internvideo2_mm import (
+        BERT_MODEL_ID,
+        INTERNVIDEO2_MODEL_FILE,
+        INTERNVIDEO2_MODEL_ID,
+        InternVideo2MultiModality,
+        _InternVideo2Stage2Wrapper,
+        _setup_internvideo2,
+    )
+except ImportError:
+    pytest.skip("InternVideo2 package not available", allow_module_level=True)
 
 # Create a random generator for consistent testing
 rng = np.random.default_rng(42)

tutorials/video/getting-started/video_split_clip_example.py

@@ -27,7 +27,17 @@
     CosmosEmbed1EmbeddingStage,
     CosmosEmbed1FrameCreationStage,
 )
-from nemo_curator.stages.video.embedding.internvideo2 import InternVideo2EmbeddingStage, InternVideo2FrameCreationStage
+
+try:
+    from nemo_curator.stages.video.embedding.internvideo2 import (
+        InternVideo2EmbeddingStage,
+        InternVideo2FrameCreationStage,
+    )
+except ImportError:
+    print("InternVideo2 is not installed")
+    InternVideo2EmbeddingStage = None
+    InternVideo2FrameCreationStage = None
+
 from nemo_curator.stages.video.filtering.clip_aesthetic_filter import ClipAestheticFilterStage
 from nemo_curator.stages.video.filtering.motion_filter import MotionFilterStage, MotionVectorDecodeStage
 from nemo_curator.stages.video.io.clip_writer import ClipWriterStage
@@ -159,6 +169,9 @@ def create_video_splitting_pipeline(args: argparse.Namespace) -> Pipeline:  # no
                 )
             )
         elif args.embedding_algorithm.startswith("internvideo2"):
+            if InternVideo2FrameCreationStage is None:
+                msg = "InternVideo2 is not installed, please consider installing it or using cosmos-embed1 instead."
+                raise ValueError(msg)
             pipeline.add_stage(
                 InternVideo2FrameCreationStage(
                     model_dir=args.model_dir,