Fix requirements (jonasrauber#51)

* removed cast no longer needed (thanks to MyPy 0.920)

* fixed version numbers of requirements to stablize tests

* upgraded tensorflow to 2.7

* fixed numpy version for dev/tests

* no longer test with python 3.6

.github/workflows/pypi.yml

@@ -12,7 +12,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v1
       with:
-        python-version: '3.6'
+        python-version: '3.7'
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip

.github/workflows/tests.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       max-parallel: 4
       matrix:
-        python-version: [3.6, 3.7]
+        python-version: [3.7]
 
     steps:
     - uses: actions/checkout@v1

eagerpy/tensor/jax.py

@@ -80,7 +80,7 @@ def unflatten(aux_data: Any, children: Sequence) -> JAXTensor:
 
             jax.tree_util.register_pytree_node(cls, flatten, unflatten)
             cls._registered = True
-        return cast(JAXTensor, super().__new__(cls))
+        return super().__new__(cls)
 
     def __init__(self, raw: "jnp.ndarray"):
         global jax

requirements-dev.txt

@@ -1,10 +1,11 @@
-flake8>=3.7.9
-black>=19.10b0
-pytest>=5.3.2
-pytest-cov>=2.8.1
-coverage>=5.0.3
-codecov>=2.0.15
-coveralls>=1.10.0
-mypy>=0.910
-pre-commit>=1.21.0
+flake8==3.8.3
+black==21.12b0
+pytest==5.4.3
+pytest-cov==2.10.0
+coverage==5.1
+codecov==2.1.12
+coveralls==3.3.1
+mypy==0.920
+pre-commit==2.5.1
 pydoc-markdown==2.0.5
+numpy==1.19.5

requirements.txt

@@ -2,4 +2,4 @@
 torch==1.9.0
 jaxlib==0.1.69
 jax==0.2.17
-tensorflow==2.6.1
+tensorflow==2.7.0

setup.py

@@ -31,7 +31,7 @@
         "Intended Audience :: Science/Research",
         "License :: OSI Approved :: MIT License",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.7",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
     keywords="",