From 326d2ae7523898df4689ebcb32f83cad9a26cdb7 Mon Sep 17 00:00:00 2001 From: Deezzir Date: Fri, 20 Sep 2024 11:26:27 -0400 Subject: [PATCH] Use unset_config for consistency --- tests/functional/test_snap_dcgm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/functional/test_snap_dcgm.py b/tests/functional/test_snap_dcgm.py index 2092136..831cdac 100644 --- a/tests/functional/test_snap_dcgm.py +++ b/tests/functional/test_snap_dcgm.py @@ -138,7 +138,7 @@ def bind_config(self, service, config, new_value): yield finally: # Revert back - self.set_config(service, config, old_value) + self.unset_config(service, config) self.check_bind_config(service, old_value) @pytest.mark.parametrize( @@ -250,7 +250,7 @@ def dependency_setup(cls): yield # Revert back - cls.set_config(cls.nv_hostengine_service, cls.nv_hostengine_port_config, old_value) + cls.unset_config(cls.nv_hostengine_service, cls.nv_hostengine_port_config) cls.check_bind_config(cls.nv_hostengine_service, old_value) subprocess.check_call(f"sudo snap start {cls.dcgm_exporter_service}".split())