added workflow env caching

.github/workflows/run_test.yaml

@@ -1,4 +1,4 @@
-name: Run Unit Test via Pytest
+name: Pytest
 
 on: [push, pull_request]
 
@@ -11,17 +11,29 @@ jobs:
 
     steps:
       - uses: actions/checkout@v3
+
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.python-version }}
+
+      - name: Cache pip packages
+        uses: actions/cache@v3
+        with:
+          path: ~/.cache/pip  # Cache pip packages
+          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/requirements.txt') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-${{ matrix.python-version }}-
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+
       - name: Test with pytest
         run: |
           coverage run -m pytest -v -s
+
       - name: Generate Coverage Report
         run: |
           coverage report -m

requirements.txt

@@ -1,6 +1,7 @@
 torch
 lightning
 torchmetrics
+rich
 fair-esm
 jsonargparse
 wandb