Skip to content

Commit 59bdb46

Browse files
committed
Update setup.py script for new pip.
1 parent 9ba448b commit 59bdb46

File tree

1 file changed

+7
-13
lines changed

1 file changed

+7
-13
lines changed

setup.py

+7-13
Original file line numberDiff line numberDiff line change
@@ -1,28 +1,22 @@
11
from setuptools import setup, find_packages
22

33

4-
try: # for pip >= 10
5-
from pip._internal.req import parse_requirements
6-
except ImportError: # for pip <= 9.0.3
7-
from pip.req import parse_requirements
4+
def parse_requirements(filename):
5+
""" load requirements from a pip requirements file """
6+
lineiter = (line.strip() for line in open(filename))
7+
return [line for line in lineiter if line and not line.startswith("#")]
88

99

10-
# parse_requirements() returns generator of pip.req.InstallRequirement objects
11-
install_reqs = parse_requirements('requirements.txt', session='null')
12-
13-
14-
# reqs is a list of requirement
15-
# e.g. ['django==1.5.1', 'mezzanine==1.4.6']
16-
reqs = [str(ir.req) for ir in install_reqs]
10+
reqs = parse_requirements('requirements.txt')
1711

1812

1913
with open('README.md') as f:
2014
long_description = f.read()
2115

2216

2317
setup(name='pytorch2keras',
24-
version='0.2.3',
25-
description='The deep learning models convertor',
18+
version='0.2.4',
19+
description='The deep learning models converter',
2620
long_description=long_description,
2721
long_description_content_type='text/markdown',
2822
url='https://github.com/nerox8664/pytorch2keras',

0 commit comments

Comments
 (0)