code: def test_dgi_save_load(tmpdir, model_type):
      base_generator, base_model, nodes = _model_data(
          model_type, sparse=False
      )
      corrupted_generator = CorruptedGenerator(base_generator)
      infomax = DeepGraphInfomax(base_model, corrupted_generator)
  
      test_utils.model_save_load(tmpdir, infomax)
  
file_path: stellargraph/tests/layer/test_deep_graph_infomax.py
imports: from stellargraph.layer import *
  from stellargraph.mapper import *
  from ..test_utils.graphs import example_graph_random
  from ..None import require_gpu, test_utils
  import tensorflow
  import pytest
  import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: test_dgi_save_load(tmpdir, model_type) without return types