@@ -18,7 +18,7 @@ def setup_mock(nvml):
1818 return nvml
1919
2020
21- @patch ('kernel_tuner.nvml.nvml' )
21+ @patch ('kernel_tuner.observers. nvml.nvml' )
2222def test_get_nvml_pwr_limits (nvml ):
2323 nvml = setup_mock (nvml )
2424 result = get_nvml_pwr_limits (0 , quiet = True )
@@ -30,7 +30,7 @@ def test_get_nvml_pwr_limits(nvml):
3030 assert result ['nvml_pwr_limit' ][- 1 ] == 150
3131
3232
33- @patch ('kernel_tuner.nvml.nvml' )
33+ @patch ('kernel_tuner.observers. nvml.nvml' )
3434def test_get_nvml_gr_clocks (nvml ):
3535 nvml = setup_mock (nvml )
3636 result = get_nvml_gr_clocks (0 , quiet = True )
@@ -40,15 +40,15 @@ def test_get_nvml_gr_clocks(nvml):
4040 assert result ['nvml_gr_clock' ] == [1000 , 3000 ]
4141
4242
43- @patch ('kernel_tuner.nvml.nvml' )
43+ @patch ('kernel_tuner.observers. nvml.nvml' )
4444def test_get_nvml_mem_clocks (nvml ):
4545 nvml = setup_mock (nvml )
4646 result = get_nvml_mem_clocks (0 , quiet = False )
4747 print (result )
4848 assert result ['nvml_mem_clock' ] == [2100 ]
4949
5050
51- @patch ('kernel_tuner.nvml.nvml' )
51+ @patch ('kernel_tuner.observers. nvml.nvml' )
5252def test_get_idle_power (nvml ):
5353 nvml = setup_mock (nvml )
5454 result = get_idle_power (0 )
0 commit comments