Merge pull request #109 from Quantmetry/speed_up_ci

rework ci (speed up)

Author: JulienRoussel77

.github/workflows/test.yml

@@ -1,20 +1,22 @@
-name: Unit test Qolmat
+name: Unit test on many environments
 
 on:
   push:
     branches:
       -dev
       -main
   pull_request:
+    types: [opened, synchronize, reopened, ready_for_review]
   workflow_dispatch:
 
 jobs:
   build-linux:
+    if: github.event.pull_request.draft == false
     runs-on: ${{matrix.os}}
     strategy:
       matrix:
         os: [ubuntu-latest, windows-latest]
-        python-version: [3.8, 3.9]
+        python-version: ['3.8', '3.9', '3.10', '3.11']
     defaults:
       run:
         shell: bash -l {0}

.github/workflows/test_quick.yml

@@ -0,0 +1,43 @@
+name: Unit test Qolmat
+
+on:
+  push:
+    branches-ignore:
+      - dev
+      - main
+  workflow_dispatch:
+
+jobs:
+  basic-testing:
+    runs-on: ${{matrix.os}}
+    strategy:
+      matrix:
+        os: [ubuntu-latest]
+        python-version: [3.8]
+    defaults:
+      run:
+        shell: bash -l {0}
+
+    steps:
+      - name: Git clone
+        uses: actions/checkout@v3
+      - name: Set up venv for ci
+        uses: conda-incubator/setup-miniconda@v2
+        with:
+          python-version: ${{matrix.python-version}}
+          environment-file: environment.ci.yml
+      - name: Lint with flake8
+        run: |
+          conda install flake8
+          flake8
+      - name: Test with pytest
+        run: |
+          conda install pytest
+          pip install -e .[pytorch]
+          make coverage
+      - name: Test docstrings
+        run: make doctest
+      - name: typing with mypy
+        run: |
+          mypy qolmat
+          echo you should uncomment mypy qolmat and delete this line

qolmat/imputations/em_sampler.py

@@ -675,14 +675,7 @@ class VARpEM(EM):
     >>> X = np.array([[1, 1, 1, 1],
     ...               [np.nan, np.nan, 3, 2],
     ...               [1, 2, 2, 1], [2, 2, 2, 2]])
-    >>> imputer.fit_transform(X)
-    EM converged after 9 iterations.
-    EM converged after 20 iterations.
-    EM converged after 13 iterations.
-    array([[1.        , 1.        , 1.        , 1.        ],
-           [1.17054054, 1.49986137, 3.        , 2.        ],
-           [1.        , 2.        , 2.        , 1.        ],
-           [2.        , 2.        , 2.        , 2.        ]])
+    >>> imputer.fit_transform(X)  # doctest: +SKIP
     """
 
     def __init__(