clean up tests

Author: apoorvalal

.github/workflows/comprehensive-tests.yml

@@ -13,15 +13,14 @@ jobs:
   test-matrix:
     name: Test Python ${{ matrix.python-version }} on ${{ matrix.os }}
     runs-on: ${{ matrix.os }}
+    timeout-minutes: 45
     strategy:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.10", "3.11", "3.12"]
         exclude:
           # Skip some combinations to reduce CI time
-          - os: macos-latest
-            python-version: "3.9"
           - os: macos-latest  
             python-version: "3.10"
 
@@ -60,7 +59,7 @@ jobs:
         
     - name: Install test dependencies
       run: |
-        pip install pyfixest pandas
+        pip install pyfixest pandas scikit-learn
         
     - name: Verify installation
       run: |
@@ -91,6 +90,7 @@ jobs:
   test-performance:
     name: Performance Tests
     runs-on: ubuntu-latest
+    timeout-minutes: 20
     steps:
     - name: Checkout code
       uses: actions/checkout@v4
@@ -105,7 +105,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install pytest pytest-benchmark
         pip install -e .
-        pip install pyfixest pandas
+        pip install pyfixest pandas scikit-learn
 
     - name: Run performance benchmarks
       run: |
@@ -136,6 +136,7 @@ jobs:
   test-examples:
     name: Test Examples and Documentation
     runs-on: ubuntu-latest
+    timeout-minutes: 25
     steps:
     - name: Checkout code
       uses: actions/checkout@v4
@@ -150,7 +151,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install jupyter nbconvert
         pip install -e .
-        pip install pyfixest pandas matplotlib seaborn
+        pip install pyfixest pandas scikit-learn matplotlib seaborn
 
     - name: Test example notebooks (if any)
       run: |

.github/workflows/fixed-effects-tests.yml

@@ -9,9 +9,10 @@ on:
 jobs:
   test:
     runs-on: ubuntu-latest
+    timeout-minutes: 30
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     steps:
     - uses: actions/checkout@v4
@@ -29,7 +30,7 @@ jobs:
 
     - name: Install test dependencies
       run: |
-        pip install pyfixest pandas
+        pip install pyfixest pandas scikit-learn
 
     - name: Run tests
       run: |
@@ -42,6 +43,7 @@ jobs:
   test-fast:
     # Quick test job for faster feedback
     runs-on: ubuntu-latest
+    timeout-minutes: 15
     steps:
     - uses: actions/checkout@v4
 
@@ -55,38 +57,8 @@ jobs:
         python -m pip install --upgrade pip
         pip install pytest
         pip install -e .
-        pip install pyfixest pandas
+        pip install pyfixest pandas scikit-learn
 
     - name: Run minimal test suite
       run: |
         python -m pytest tests/test_linear.py tests/test_fe.py::test_minimal_example -v
-
-  lint:
-    runs-on: ubuntu-latest
-    steps:
-    - uses: actions/checkout@v4
-
-    - name: Set up Python
-      uses: actions/setup-python@v4
-      with:
-        python-version: "3.11"
-
-    - name: Install linting tools
-      run: |
-        python -m pip install --upgrade pip
-        pip install black flake8 isort
-
-    - name: Check code formatting with black
-      run: |
-        black --check --diff .
-
-    - name: Check import sorting with isort
-      run: |
-        isort --check-only --diff .
-
-    - name: Lint with flake8
-      run: |
-        # Stop the build if there are Python syntax errors or undefined names
-        flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
-        # Exit-zero treats all errors as warnings. Line length set to 88 (black default)
-        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=88 --statistics