github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/1278/head
Jiarui Fang 2022-07-12 23:04:06 +08:00 committed by GitHub
parent 7aadcbd070
commit ca9d5ee91c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_fx/test_pipeline/test_torchvision/test_torchvision.py
View File

@ -10,6 +10,7 @@ from torch.fx import GraphModule
import random import random
import numpy as np import numpy as np
import inspect import inspect
import pytest
MANUAL_SEED = 0 MANUAL_SEED = 0
random.seed(MANUAL_SEED) random.seed(MANUAL_SEED)