From 7ef42a2f06ae8a6bd079e0b6664c951b5c18eb63 Mon Sep 17 00:00:00 2001
From: xiongkun <xiongkun5@huawei.com>
Date: Fri, 1 Apr 2022 15:56:11 +0800
Subject: [PATCH] [MS][rewrite]add required_package

---
 requirements.txt | 1 -
 setup.py         | 3 ++-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 28e7297e0aa..e8fec0f291b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,4 +13,3 @@ tables >= 3.6.1             # for st test
 easydict >= 1.9             # for st test
 psutil >= 5.7.0
 astunparse >= 0.0
-astpretty >= 0.0
diff --git a/setup.py b/setup.py
index b1581b1c499..63d69b005cf 100644
--- a/setup.py
+++ b/setup.py
@@ -114,7 +114,8 @@ required_package = [
     'pillow >= 6.2.0',
     'scipy >= 1.5.2',
     'packaging >= 20.0',
-    'psutil >= 5.6.1'
+    'psutil >= 5.6.1',
+    'astunparse >= 0.0'
 ]
 
 package_data = {