Require torch>=1.6; drop support for Python 3.5 (#380)

pyro-ppl · Oct 13, 2020 · 330a07d · 330a07d
1 parent 778e70a
commit 330a07d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 14 deletions.
diff --git 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
@@ -32,7 +32,7 @@
     },
     author='Uber AI Labs',
     author_email='[email protected]',
-    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,14 +72,15 @@
     },
     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',
         'Intended Audience :: Science/Research',
         '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',
     ],
 )