Fix tests related to changing default gpu_device_plugin_yaml

astronomer · Dec 2, 2024 · b9378bf · b9378bf
1 parent 7c20206
commit b9378bf
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/test_operators.py b/tests/test_operators.py
@@ -35,10 +35,7 @@ def test_init_default_values(self):
             ray_cluster_yaml="cluster.yaml",
         )
         assert operator.kuberay_version == "1.0.0"
-        assert (
-            operator.gpu_device_plugin_yaml
-            == "https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.9.0/nvidia-device-plugin.yml"
-        )
+        assert not operator.gpu_device_plugin_yaml
         assert operator.update_if_exists is False
 
     @patch("ray_provider.operators.RayHook")
@@ -87,10 +84,7 @@ def test_init_default_gpu_plugin(self):
             conn_id="test_conn",
             ray_cluster_yaml="cluster.yaml",
         )
-        assert (
-            operator.gpu_device_plugin_yaml
-            == "https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.9.0/nvidia-device-plugin.yml"
-        )
+        assert not operator.gpu_device_plugin_yaml
 
     @patch("ray_provider.operators.RayHook")
     def test_hook_property(self, mock_ray_hook):