From b50ad2a522d8d629aaa4ac3f977079062d1140ae Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Wed, 19 May 2021 16:38:21 +0800
Subject: [PATCH] Fix bugs in test

---
 README.md         | 5 ++++-
 tests/test_tas.py | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 6255b3e..db2fd09 100644
--- a/README.md
+++ b/README.md
@@ -88,7 +88,10 @@ At this moment, this project provides the following algorithms and scripts to ru
 
 ## Requirements and Preparation
 
-Please install `Python>=3.6` and `PyTorch>=1.5.0`. (You could also run this project in lower versions of Python and PyTorch, but may have bugs).
+
+**First of all**, please use `python setup.py install` to install `xautodl` library.
+
+Please install `Python>=3.6` and `PyTorch>=1.5.0`. (You could use lower versions of Python and PyTorch, but may have bugs).
 Some visualization codes may require `opencv`.
 
 CIFAR and ImageNet should be downloaded and extracted into `$TORCH_HOME`.
diff --git a/tests/test_tas.py b/tests/test_tas.py
index bb46313..53f3702 100644
--- 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