rename wheel

.travis.yml

 language: shell
 
 os:
-  # - linux
+  - linux
   # - osx
   - windows
 
@@ -9,7 +9,7 @@ env:
   global:
     - CUDA_HOME=/usr/local/cuda
   jobs:
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101
@@ -67,6 +67,7 @@ script:
   - python setup.py test
 after_success:
   - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}/${IDX}
+  - python script/rename_wheel.py
   - codecov
 deploy:
   provider: s3

script/rename_wheel.py

+import os
+import os.path as osp
+import glob
+
+dist_dir = osp.join(osp.dirname(osp.abspath(__file__)), '..', 'dist')
+wheels = glob.glob(osp.join('dist', '**', '*.whl'), recursive=True)
+
+for wheel in wheels:
+    idx = wheel.split(osp.sep)[-2]
+    if idx not in ['cpu', 'cu92', 'cu100', 'cu101']:
+        continue
+    name = wheel.split(osp.sep)[-1]
+    if idx in name:
+        continue
+
+    names = name.split('-')
+    name = '-'.join(names[:-4] + [names[-4] + '%2B' + idx] + names[-3:])
+    new_wheel = osp.join(*wheel.split(osp.sep)[:-1], name)
+    os.rename(wheel, new_wheel)