code: def test_wide_deep(notebooks, output_notebook, kernel_name, tmp):
      notebook_path = notebooks["wide_deep"]
  
      # Simple test (train only 1 batch == 1 step)
      model_dir = os.path.join(tmp, "wide_deep_0")
      os.mkdir(model_dir)
      params = {
          "MOVIELENS_DATA_SIZE": "mock100",
          "STEPS": 1,
          "EVALUATE_WHILE_TRAINING": False,
          "MODEL_DIR": model_dir,
          "EXPORT_DIR_BASE": model_dir,
          "RATING_METRICS": ["rmse"],
          "RANKING_METRICS": ["ndcg_at_k"],
      }
      execute_notebook(
          notebook_path,
          output_notebook,
          kernel_name=kernel_name,
          parameters=params,
      )
  
      # Test with different parameters
      model_dir = os.path.join(tmp, "wide_deep_1")
      os.mkdir(model_dir)
      params = {
          "MOVIELENS_DATA_SIZE": "mock100",
          "STEPS": 1,
          "ITEM_FEAT_COL": None,
          "EVALUATE_WHILE_TRAINING": True,
          "MODEL_DIR": model_dir,
          "EXPORT_DIR_BASE": model_dir,
          "RATING_METRICS": ["rsquared"],
          "RANKING_METRICS": ["map_at_k"],
      }
      execute_notebook(
          notebook_path,
          output_notebook,
          kernel_name=kernel_name,
          parameters=params,
      )
  
file_path: recommenders/tests/unit/examples/test_notebooks_gpu.py
imports: import os
  import pytest
  from recommenders.utils.gpu_utils import get_number_gpus
  from recommenders.utils.notebook_utils import execute_notebook
label_desc: Use image node_id 6 for calling a global function with example usage: test_wide_deep(notebooks, output_notebook, kernel_name, tmp) without return types