Revert "Skip internvideo2 Test if not installed (#1089)" (#1110)

This reverts commit b852ef7.

Author: thomasdhc

.gitignore

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

external/intern_video2_installation.sh

@@ -18,17 +18,12 @@
 from unittest.mock import MagicMock, Mock, patch
 
 import numpy as np
-import pytest
 import torch
 
-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.stages.video.embedding.internvideo2 import (
+    InternVideo2EmbeddingStage,
+    InternVideo2FrameCreationStage,
+)
 from nemo_curator.tasks.video import Clip, Video, VideoTask
 
 # Create a random generator for consistent testing

tests/models/test_internvideo2.py

@@ -21,17 +21,14 @@
 import torch
 from easydict import EasyDict
 
-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)
+from nemo_curator.models.internvideo2_mm import (
+    BERT_MODEL_ID,
+    INTERNVIDEO2_MODEL_FILE,
+    INTERNVIDEO2_MODEL_ID,
+    InternVideo2MultiModality,
+    _InternVideo2Stage2Wrapper,
+    _setup_internvideo2,
+)
 
 # Create a random generator for consistent testing
 rng = np.random.default_rng(42)

tests/models/test_internvideo2_mm.py

@@ -27,17 +27,7 @@
     CosmosEmbed1EmbeddingStage,
     CosmosEmbed1FrameCreationStage,
 )
-
-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.embedding.internvideo2 import InternVideo2EmbeddingStage, InternVideo2FrameCreationStage
 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
@@ -169,9 +159,6 @@ 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,