better testing

Author: SkafteNicki

.github/workflows/flake8.yml

@@ -14,10 +14,10 @@ jobs:
     steps:
     - name: Checkout
       uses: actions/checkout@v2
-    - name: Set up Python 3.8
+    - name: Set up Python 3.9
       uses: actions/setup-python@v2
       with:
-        python-version: 3.8
+        python-version: 3.9
     - name: Install flake8
       run: pip install flake8
     - name: Run flake8

.github/workflows/isort.yml

@@ -13,10 +13,10 @@ jobs:
 
     steps:
     - uses: actions/checkout@v2
-    - name: Set up Python 3.8
+    - name: Set up Python 3.9
       uses: actions/setup-python@v2
       with:
-        python-version: 3.8
+        python-version: 3.9
     - name: Install isort
       run: pip install isort==5.6.4
 

.github/workflows/tests.yml

@@ -9,24 +9,34 @@ on:
 jobs:
   build:
 
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-20.04, macOS-10.15, windows-2019]
+        python-version: [3.6, 3.7, 3.8, 3.9]
+        pytorch-version: ["1.6", "1.7", "1.8", "1.9", "1.10", "1.11"]
 
     steps:
     - name: Checkout
       uses: actions/checkout@v2
-    - name: Set up Python 3.8
+
+    - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v2
       with:
-        python-version: 3.8
+        python-version: ${{ matrix.python-version }}
+
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
+        pip install torch==${{ matrix.pytorch-version }}
         pip install -r requirements.txt
         pip install -r requirements_tests.txt
+
     - name: Install package
       run: |
         python setup.py install sdist bdist_wheel
         twine check dist/*
+
     - name: Test with pytest
       run: |
         pip install pytest coverage

README.md

@@ -4,6 +4,7 @@
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pytorch-lightning)](https://pypi.org/project/stochman/)
 [![PyPI Status](https://badge.fury.io/py/stochman.svg)](https://badge.fury.io/py/stochman)
 [![PyPI Status](https://pepy.tech/badge/stochman)](https://pepy.tech/badge/stochman)
+[![license](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/MachineLearningLifeScience/stochman/blob/master/LICENSE)
 
 # StochMan - Stochastic Manifolds made easier