refactor(//tests/py): Apply linting to py tests

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

tests/py/test_multi_gpu.py

@@ -5,7 +5,9 @@
 
 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 @@ def test_compile_script(self):
         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()