Add requirements in setup

.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
 

.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

.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

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*")