mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-18 07:57:46 +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 numpy as np
|
||||
import inspect
|
||||
import pytest
|
||||
|
||||
MANUAL_SEED = 0
|
||||
random.seed(MANUAL_SEED)
|
||||
|
Loading…
Reference in New Issue
Block a user