Skip to content

Commit a0f61e6

Browse files
committed
edit import in test image recog model
1 parent 99e96f9 commit a0f61e6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/test_image_recog_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import torch
2-
from onepiece_classify.models.build_model import image_recog
2+
from onepiece_classify.models import image_recog
33

44
def test_model():
55
rand_tensor = torch.rand([1, 3, 224, 224])

0 commit comments

Comments
 (0)