Same as `test_jit.py` this needs to still run if someone doesn't have torchvision installed. Follow the pattern there, but it would be something like:    ```  try:      import torchvision      HAS_TORCHVISION = True  except ImportError:      HAS_TORCHVISION = False      skipIfNoTorchVision = unittest.skipIf(not HAS_TORCHVISION, "no torchvision")  ```