Skip to content

Commit

Permalink
Update setup.py script for new pip.
Browse files Browse the repository at this point in the history
  • Loading branch information
gmalivenko committed May 14, 2020
1 parent 9ba448b commit 59bdb46
Showing 1 changed file with 7 additions and 13 deletions.
20 changes: 7 additions & 13 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,28 +1,22 @@
from setuptools import setup, find_packages


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


# parse_requirements() returns generator of pip.req.InstallRequirement objects
install_reqs = parse_requirements('requirements.txt', session='null')


# reqs is a list of requirement
# e.g. ['django==1.5.1', 'mezzanine==1.4.6']
reqs = [str(ir.req) for ir in install_reqs]
reqs = parse_requirements('requirements.txt')


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


setup(name='pytorch2keras',
version='0.2.3',
description='The deep learning models convertor',
version='0.2.4',
description='The deep learning models converter',
long_description=long_description,
long_description_content_type='text/markdown',
url='https://github.com/nerox8664/pytorch2keras',
Expand Down

0 comments on commit 59bdb46

Please sign in to comment.