Use unset_config for consistency

canonical · Sep 20, 2024 · 326d2ae · 326d2ae
1 parent d56b699
commit 326d2ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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())