run unittests only for tests/unit folder

Author: sfluegel

.github/workflows/test.yml

@@ -24,4 +24,4 @@ jobs:
           python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           python -m pip install -e .
       - name: Display Python version
-        run: python -m unittest
+        run: python -m unittest discover -s tests/unit