Fix bugs in test

2021-05-19 16:38:21 +08:00
parent a46095c2ee
commit b50ad2a522
2 changed files with 5 additions and 1 deletions
--- a/tests/test_tas.py
+++ b/tests/test_tas.py
@@ -3,6 +3,7 @@
 ##################################################
 import torch
 import torch.nn as nn
+import unittest

 from xautodl.models.shape_searchs.SoftSelect import ChannelWiseInter