diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 1956420..a8fa38f 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -32,6 +32,7 @@ jobs:
         # Installing pytorch on cpu and torchvision for running tests 
         python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
         python -m pip install matplotlib
+        python -m pip install .
     - name: Lint with flake8
       run: |
         # stop the build if there are Python syntax errors or undefined names