diff --git a/.travis.yml b/.travis.yml index 1c631ee53..5a24863c1 100644 --- a/.travis.yml +++ b/.travis.yml @@ -33,7 +33,7 @@ jobs: - name: torch python: 3.6 script: - - pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html + - pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html # Keep track of Pyro dev branch - pip install https://github.com/pyro-ppl/pyro/archive/dev.zip - pip install -e .[torch] @@ -45,11 +45,3 @@ jobs: - pip install https://github.com/pyro-ppl/numpyro/archive/master.zip - pip install -e .[jax] - CI=1 FUNSOR_BACKEND=jax make test - - name: torch35 - python: 3.5 - script: - - pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html - # Keep track of Pyro dev branch - - pip install https://github.com/pyro-ppl/pyro/archive/dev.zip - - pip install -e .[torch] - - FUNSOR_BACKEND=torch make test diff --git a/setup.py b/setup.py index 05cd15668..756d595d2 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,7 @@ }, author='Uber AI Labs', author_email='fritzo@uber.com', - python_requires=">=3.5", + python_requires=">=3.6", install_requires=[ 'makefun', 'multipledispatch', @@ -41,8 +41,8 @@ ], extras_require={ 'torch': [ - 'pyro-ppl>=0.5', - 'torch>=1.3.0', + 'pyro-ppl>=1.4', + 'torch>=1.6.0', ], 'jax': [ 'numpyro>=0.2.4', @@ -72,7 +72,7 @@ }, long_description=long_description, long_description_content_type='text/markdown', - keywords='probabilistic machine learning bayesian statistics pytorch', + keywords='probabilistic machine learning bayesian statistics pytorch jax', classifiers=[ 'Intended Audience :: Developers', 'Intended Audience :: Education', @@ -80,6 +80,7 @@ 'License :: OSI Approved :: Apache Software License', 'Operating System :: POSIX :: Linux', 'Operating System :: MacOS :: MacOS X', - 'Programming Language :: Python :: 3.5', + 'Programming Language :: Python :: 3.6', + 'Programming Language :: Python :: 3.7', ], )