code: def test_enable_disable_nvml():
      with dask.config.set({"distributed.diagnostics.nvml": False}):
          nvml.init_once()
          assert not nvml.is_initialized()
          assert nvml.NVML_STATE == nvml.NVMLState.DISABLED_CONFIG
  
      # Idempotent (once we've decided not to turn things on with
      # configuration, it's set in stone)
      nvml.init_once()
      assert not nvml.is_initialized()
      assert nvml.NVML_STATE == nvml.NVMLState.DISABLED_CONFIG
  
file_path: distributed/distributed/diagnostics/tests/test_nvml.py
imports: from __future__ import annotations
  import multiprocessing
  import os
  import pytest
  import dask
  from distributed.diagnostics import nvml
  from distributed.utils_test import gen_cluster
label_desc: Use image node_id 3 for calling a global function with example usage: test_enable_disable_nvml() without return types