Fix pipeline

.github/workflows/main.yaml

@@ -2,11 +2,17 @@ name: lab-testing
 
 # add your pipline description below
 
-
 on:
+  push:
+    branches:
+      - main
   pull_request:
+    branches:
+      - main
+
+
 jobs:
-      test:
+  unit-tests:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -20,10 +26,8 @@ on:
       - name: Install dependencies
         run: poetry install; poetry add pytest; poetry add pytest-mock;
       - name: Run pytest
-            run: poetry run pytest tests/e2e/test_*.py
+        run: poetry run pytest tests/unit/test_*.py
-  push:
+  e2e-tests:
-    jobs:
-      test:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -37,4 +41,4 @@ on:
       - name: Install dependencies
         run: poetry install; poetry add pytest; poetry add pytest-mock;
       - name: Run pytest
-            run: poetry run pytest tests/unit/test_*.py
+        run: poetry run pytest tests/e2e/test_*.py