Add python 3.9 & 3.10 supports and Drop 3.6

Author: eelregit

.github/workflows/pythonpackage.yml

@@ -15,7 +15,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.6, 3.7, 3.8]
+        python-version: ['3.7', '3.8', '3.9', '3.10']
 
     steps:
     - uses: actions/checkout@v2
@@ -32,10 +32,10 @@ jobs:
       shell: bash -l {0}
       run: |
         conda config --set always_yes yes
-        conda install pytest pip 
+        conda install pytest pip
         conda install -c conda-forge pyccl
         pip install .
-  
+
     - name: Test with pytest
       shell: bash -l {0}
       run: |

setup.py

@@ -17,14 +17,16 @@
     author="jax-cosmo developers",
     packages=find_packages(),
     url="https://github.com/DifferentiableUniverseInitiative/jax_cosmo",
+    python_requires=">=3.7",
     install_requires=["jax", "jaxlib"],
     tests_require=["pyccl"],
     use_scm_version=True,
     setup_requires=["setuptools_scm"],
     classifiers=[
-        "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
         "License :: OSI Approved :: MIT License",
         "Operating System :: MacOS",
         "Operating System :: POSIX :: Linux",