From b9378bfb7a4b89e06db39870670819ac2d0b8bc2 Mon Sep 17 00:00:00 2001 From: Tatiana Al-Chueyr Date: Mon, 2 Dec 2024 11:59:06 +0000 Subject: [PATCH] Fix tests related to changing default gpu_device_plugin_yaml --- tests/test_operators.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/tests/test_operators.py b/tests/test_operators.py index a351f3b..faa89e3 100644 --- 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):