diff --git a/providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_ray.py b/providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_ray.py index 20e5e8829d662..c4905cc1c52c8 100644 --- a/providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_ray.py +++ b/providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_ray.py @@ -19,6 +19,9 @@ from unittest import mock +import pytest + +resources = pytest.importorskip("google.cloud.aiplatform.vertex_ray.util.resources") from google.cloud.aiplatform.vertex_ray.util.resources import Cluster, Resources from airflow.providers.google.cloud.hooks.vertex_ai.ray import RayHook