message: refactor(//tests/py): Apply linting to py tests
diff: @@ -5,7 +5,9 @@ import torchvision.models as models
  from model_test_case import ModelTestCase
  +
  class TestMultiGpuSwitching(ModelTestCase):
  +
  def setUp(self):
  if torch.cuda.device_count() < 2:
  self.fail("Test is not relevant for this platform since number of available CUDA devices is less than 2")
  @@ -55,12 +57,14 @@ class TestMultiGpuSwitching(ModelTestCase):
  trtorch.set_device(0)
  self.assertTrue(same < 2e-3)
  +
  def test_suite():
  suite = unittest.TestSuite()
  suite.addTest(TestMultiGpuSwitching.parametrize(TestMultiGpuSwitching, model=models.resnet18(pretrained=True)))
  return suite
  +
  suite = test_suite()
  runner = unittest.TextTestRunner()
  