diff --git a/action.yml b/action.yml
index 7a8b7d7..da35cb8 100644
--- a/action.yml
+++ b/action.yml
@@ -144,6 +144,18 @@ runs:
         github_token: ${{ inputs.github_token }}
         workdir: ${{ inputs.root }}/${{ inputs.workdir }}
 
+    - if: ${{ inputs.pylint == 'true' }}
+      uses: actions/setup-python@v4
+      with:
+        python-version: ${{ inputs.python_version }}
+        cache: 'pip' # caching pip dependencies
+    - if: ${{ inputs.pytest == 'true' }}
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install pytest pylint
+        cd "${{ inputs.root }}/${{ inputs.workdir }}"
+        python -m flit install || python -m pip install .
+
     - if: ${{ inputs.pylint == 'true' }}
       uses: dciborow/action-pylint@0.1.0
       with: