[hotfix] torchvison fx unittests miss import pytest (#1277)

2 years ago · ca9d5ee91c
1 changed files with 1 additions and 0 deletions
--- a/tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py
+++ b/tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py
@ -10,6 +10,7 @@ from torch.fx import GraphModule
 import random
 import numpy as np
 import inspect
+import pytest

 MANUAL_SEED = 0
 random.seed(MANUAL_SEED)