Commit 05d1cbe7 authored by Yuxin Wu's avatar Yuxin Wu

fix setuptools environment marker issue

parent ca7bc07f
...@@ -4,6 +4,7 @@ assert version > 30, "tensorpack installation requires setuptools > 30" ...@@ -4,6 +4,7 @@ assert version > 30, "tensorpack installation requires setuptools > 30"
from setuptools import setup from setuptools import setup
import os import os
import shutil import shutil
import sys
# setup metainfo # setup metainfo
CURRENT_DIR = os.path.dirname(__file__) CURRENT_DIR = os.path.dirname(__file__)
...@@ -22,6 +23,8 @@ reqfile = os.path.join(CURRENT_DIR, 'requirements.txt') ...@@ -22,6 +23,8 @@ reqfile = os.path.join(CURRENT_DIR, 'requirements.txt')
req = [x.strip() for x in open(reqfile).readlines()] req = [x.strip() for x in open(reqfile).readlines()]
reqfile = os.path.join(CURRENT_DIR, 'opt-requirements.txt') reqfile = os.path.join(CURRENT_DIR, 'opt-requirements.txt')
extra_req = [x.strip() for x in open(reqfile).readlines()] extra_req = [x.strip() for x in open(reqfile).readlines()]
if sys.version_info.major < 3:
extra_req.append('tornado')
# parse scripts # parse scripts
scripts = ['scripts/plot-point.py', 'scripts/dump-model-params.py'] scripts = ['scripts/plot-point.py', 'scripts/dump-model-params.py']
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment