code: def test_mnist_pytorch_distributed(image_type: str) -> None:
      config = conf.load_config(
          conf.tutorials_path("mnist_pytorch/distributed.yaml")
      )
      config = conf.set_max_length(config, {"batches": 200})
  
      if image_type == "PT":
          config = conf.set_pt_image(config)
      elif image_type == "PT2":
          config = conf.set_pt2_image(config)
      elif image_type == "TF2":
          config = conf.set_tf2_image(config)
      else:
          warnings.warn("Using default images", stacklevel=2)
  
      exp.run_basic_test_with_temp_config(
          config, conf.tutorials_path("mnist_pytorch"), 1
      )
  
file_path: determined/e2e_tests/tests/nightly/test_distributed.py
imports: import os
  import shutil
  import tempfile
  import warnings
  import pytest
  from determined.common import util
  from tests import config
  from tests import experiment
label_desc: Use image node_id 1 for calling a global function with example usage: test_mnist_pytorch_distributed(image_type) without return types