code: def test_plots_on_gpu():
      """If there is GPU available running all the docs plot files. Only makes sure the plots don't crash, and not \
      testing any other display or functionality."""
      if torch.cuda.is_available():
          path = (
              Path(__file__).parent.parent.parent.parent
              / "docs"
              / "source"
          )
          # Take only source file and excluding compiled files
          source_files = set()
          for folder in DOCS_EXAMPLES_DIR:
              source_files.update(
                  set(path.glob(f"**/{folder}/**/plot_*.py"))
              )
  
          if not source_files:
              raise ValueError("No plots found in docs/source")
          for file in source_files:
              print(f"plot file: {str(file)}")
              start = time.time()
              run_path(str(file))
              end = time.time()
              print(f"plot file: {str(file)}, Time: {end - start}")
  
imports: import time
  from pathlib import Path
  from runpy import run_path
  import torch
  import wandb
label_desc: Use image node_id 1 for calling a global function with example usage: test_plots_on_gpu() without return types