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

ca9d5ee9 · Jiarui Fang · GitHub · 7aadcbd0 · ca9d5ee9
Unverified Commit ca9d5ee9 authored Jul 12, 2022 by Jiarui Fang Committed by GitHub Jul 12, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py ...est_fx/test_pipeline/test_torchvision/test_torchvision.py +1 -0

No files found.
--- 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)