add cuda tagging for test

Author: cyruszhang

tests/ops/mapper/test_image_sam_3d_body_mapper.py

@@ -17,7 +17,7 @@ def _is_egl_available():
         from OpenGL.platform import ctypesloader
         ctypesloader.loadLibrary(None, 'EGL')
         return True
-    except (ImportError, OSError):
+    except (ImportError, OSError, TypeError):
         return False
 
 

tests/ops/mapper/test_text_tagging_by_prompt_mapper.py

@@ -1,6 +1,7 @@
 import unittest
 from data_juicer.ops.mapper.text_tagging_by_prompt_mapper import TextTaggingByPromptMapper, DEFAULT_CLASSIFICATION_PROMPT, DEFAULT_CLASSIFICATION_LIST
 from data_juicer.utils.constant import Fields
+from data_juicer.utils.resource_utils import is_cuda_available
 from data_juicer.utils.unittest_utils import DataJuicerTestCaseBase
 
 def check_string_in_list(string_list, output):
@@ -40,6 +41,7 @@ def test_tagging(self):
             }]
         self._run_tagging(samples)
 
+    @unittest.skipUnless(is_cuda_available(), 'vLLM requires CUDA')
     def test_tagging_vllm(self):
         samples = [
             {