code: def test_pytorch_distributed_with_amp(
      self, tmp_path: pathlib.Path, api_style
  ):
      launch_config = pytorch_utils.setup_torch_distributed()
  
      outputs = launcher.elastic_launch(launch_config, run_amp)(
          tmp_path, api_style
      )
      launcher.elastic_launch(launch_config, run_amp)(
          tmp_path, api_style, outputs[0]
      )
  
imports: import contextlib
  import importlib
  import io
  import os
  import pathlib
  import sys
  import typing
  from unittest import mock
  import numpy
  import pytest
  import torch
  from _pytest import monkeypatch
  from torch.distributed import launcher
  import determined
  from determined import pytorch
  from tests.experiment import pytorch_utils, utils
  from tests.experiment.fixtures import pytorch_onevar_model
label_desc: Use image node_id 28 for calling the TestPyTorchTrial obj's underlying member method code with example usage: obj.test_pytorch_distributed_with_amp(tmp_path, api_style) without return types