Skip tests if dependencies is missing

Author: AMOOOMA

sdks/python/apache_beam/ml/inference/model_manager_it_test.py

@@ -1,8 +1,10 @@
 import unittest
-import torch
 import apache_beam as beam
 from apache_beam.ml.inference.base import RunInference
+
+# pylint: disable=ungrouped-imports
 try:
+  import torch
   from apache_beam.ml.inference.huggingface_inference import HuggingFacePipelineModelHandler
 except ImportError as e:
   raise unittest.SkipTest(

sdks/python/apache_beam/ml/inference/model_manager_test.py

@@ -5,7 +5,10 @@
 from concurrent.futures import ThreadPoolExecutor
 from unittest.mock import patch
 
-from apache_beam.ml.inference.model_manager import ModelManager, GPUMonitor, ResourceEstimator
+try:
+  from apache_beam.ml.inference.model_manager import ModelManager, GPUMonitor, ResourceEstimator
+except ImportError as e:
+  raise unittest.SkipTest("Model Manager dependencies are not installed")
 
 
 class MockGPUMonitor: