mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-18 16:07:17 +00:00
[hotfix] torchvison fx unittests miss import pytest (#1277)
This commit is contained in:
parent
7aadcbd070
commit
ca9d5ee91c
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user