Method setup_method

tests/test_gpu.py:196–204

(self)

class TestGpuTracking:
    def setup_method(self):
        for module_name in list(sys.modules.keys()):
            if module_name.startswith("codecarbon.core.gpu"):
                sys.modules.pop(module_name, None)
        import codecarbon.core

        for attr in ["gpu", "gpu_nvidia", "gpu_amd", "gpu_device"]:
            if hasattr(codecarbon.core, attr):
                delattr(codecarbon.core, attr)

    @mock.patch("codecarbon.core.gpu.is_rocm_system", return_value=True)
    @mock.patch("codecarbon.core.gpu.is_nvidia_system", return_value=False)