From f96ecfdeeea9985b92237f39f339a6bbf210fdae Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Fri, 11 Jun 2021 12:01:58 +0800
Subject: [PATCH] Add requirements in setup

---
 .github/workflows/test-basic.yml             | 6 +-----
 .github/workflows/test-misc.yml              | 2 +-
 .github/workflows/test-super-layer_model.yml | 2 +-
 setup.py                                     | 2 +-
 4 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/test-basic.yml b/.github/workflows/test-basic.yml
index a3396c6..879da72 100644
--- a/.github/workflows/test-basic.yml
+++ b/.github/workflows/test-basic.yml
@@ -45,7 +45,7 @@ jobs:
 
       - name: Test Search Space
         run: |
-          python -m pip install pytest numpy
+          python -m pip install pytest
           python -m pip install torch torchvision
           python -m pip install parameterized
           echo $PWD
@@ -58,10 +58,6 @@ jobs:
 
       - name: Test Math
         run: |
-          python -m pip install pytest numpy
-          python -m pip install parameterized
-          python -m pip install torch torchvision
-          python --version
           python -m pytest ./tests/test_math*.py -s
         shell: bash
 
diff --git a/.github/workflows/test-misc.yml b/.github/workflows/test-misc.yml
index ece0105..bfea43e 100644
--- a/.github/workflows/test-misc.yml
+++ b/.github/workflows/test-misc.yml
@@ -30,7 +30,7 @@ jobs:
 
       - name: Test Xmisc
         run: |
-          python -m pip install pytest numpy
+          python -m pip install pytest
           python -m pip install torch torchvision
           python -m pip install parameterized
           echo $PWD
diff --git a/.github/workflows/test-super-layer_model.yml b/.github/workflows/test-super-layer_model.yml
index b94440f..eef2420 100644
--- a/.github/workflows/test-super-layer_model.yml
+++ b/.github/workflows/test-super-layer_model.yml
@@ -30,7 +30,7 @@ jobs:
 
       - name: Test Super Model
         run: |
-          python -m pip install pytest numpy
+          python -m pip install pytest
           python -m pip install parameterized
           python -m pip install torch torchvision
           python -m pytest ./tests/test_super_*.py
diff --git a/setup.py b/setup.py
index 943dd4b..ea078f1 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@ def read(fname="README.md"):
 
 
 # What packages are required for this module to be executed?
-REQUIRED = ["numpy>=1.16.5,<=1.19.5"]
+REQUIRED = ["numpy>=1.16.5,<=1.19.5", "pyyaml>=5.0.0"]
 
 packages = find_packages(
     exclude=("tests", "scripts", "scripts-search", "lib*", "exps*")