From ca9d5ee91c20830560827dd7f66624d86f564ae7 Mon Sep 17 00:00:00 2001 From: Jiarui Fang Date: Tue, 12 Jul 2022 23:04:06 +0800 Subject: [PATCH] [hotfix] torchvison fx unittests miss import pytest (#1277) --- tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py b/tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py index dab485063..e52889e3b 100644 --- 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)