code: def test_wrap_scaler(self) -> None:
      scaler = torch.cuda.amp.GradScaler()  # type: ignore # GradScaler.__init__ is untyped
      assert scaler == self.context.wrap_scaler(scaler)
      assert scaler == self.context._scaler
  
file_path: determined/harness/tests/experiment/pytorch/test_pytorch_context.py
imports: import pytest
  import torch
  import determined
  from determined import errors, pytorch
label_desc: Use image node_id 4 for calling the TestPyTorchContext obj's underlying member method code with example usage: obj.test_wrap_scaler() without return types