Fix: SYS_ADMIN warning in PLO tests

ml-energy · Jul 12, 2024 · 5526da2 · 5526da2
1 parent f46b160
commit 5526da2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/optimizer/test_power_limit_optimizer.py b/tests/optimizer/test_power_limit_optimizer.py
@@ -23,6 +23,7 @@
 import pytest
 import pandas as pd
 
+from zeus.device.gpu import get_gpus
 from zeus.optimizer.power_limit import (
     GlobalPowerLimitOptimizer,
     HFGlobalPowerLimitOptimizer,
@@ -157,6 +158,9 @@ def test_power_limit_optimizer(
     # Test JIT profiling
     ############################
 
+    # Disable `SYS_ADMIN` capability check.
+    get_gpus()._disable_sys_admin_warning = True
+
     plo = GlobalPowerLimitOptimizer(
         monitor,
         optimum_selector=ZeusCost(