fix tests

aniketmaurya · aniketmaurya · commit 994a6d38aa1b · 2024-08-30T20:15:37.000+01:00
diff --git a/tests/test_connector.py b/tests/test_connector.py
@@ -88,6 +88,7 @@ def test_check_cuda_with_nvidia_smi_mock_gpu(mock_subprocess):
     ],
 )
 def test_connector(input_accelerator, expected_accelerator, expected_devices):
+    check_cuda_with_nvidia_smi.cache_clear()
     connector = _Connector(accelerator=input_accelerator)
     assert (
         connector.accelerator == expected_accelerator

Original file line number	Diff line number	Diff line change
`@@ -88,6 +88,7 @@ def test_check_cuda_with_nvidia_smi_mock_gpu(mock_subprocess):`
`88`	`88`	`],`
`89`	`89`	`)`
`90`	`90`	`def test_connector(input_accelerator, expected_accelerator, expected_devices):`
	`91`	`+ check_cuda_with_nvidia_smi.cache_clear()`
`91`	`92`	`connector = _Connector(accelerator=input_accelerator)`
`92`	`93`	`assert (`
`93`	`94`	`connector.accelerator == expected_accelerator`