code: def test_input_checks(self) -> None:
      model = torch.nn.Linear(2, 2)
  
      with self.assertRaisesRegex(
          ValueError, "Decay must be between 0 and 1"
      ):
          AveragedModel(model, averaging_method="ema", ema_decay=1.3)
  
      with self.assertRaisesRegex(
          ValueError,
          "Unknown averaging method: foo. Only ema and swa are supported.",
      ):
          # pyre-ignore On purpose to test run time exception
          AveragedModel(model, averaging_method="foo")
  
file_path: tnt/tests/utils/test_swa.py
imports: import itertools
  import unittest
  from copy import deepcopy
  from typing import List, Tuple
  import torch
  from torchtnt.utils.swa import AveragedModel
label_desc: Use image node_id 8 for calling the TestSWA obj's underlying member method code with example usage: obj.test_input_checks() without return types