Spaces:
Sleeping
Sleeping
File size: 1,486 Bytes
29b9c56 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | import os
import re
from setuptools import setup, find_packages
# Utility function to read the README file.
# Used for the long_description. It's nice, because now 1) we have a top level
# README file and 2) it's easier to type in the README file than to put a raw
# string in below ...
def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
# Read metadata from version file
metadata_file = open(os.path.join(os.path.dirname(__file__), 'pytorch_wavelets', '_version.py')).read()
metadata = dict(re.findall("__([a-z]+)__ = '([^']+)'", metadata_file))
setup(
name='pytorch_wavelets',
version=metadata['version'],
author="Fergal Cotter",
author_email="fbc23@cam.ac.uk",
description=("A port of the DTCWT toolbox to run on pytorch"),
license="Free To Use",
keywords="pytorch, DWT, DTCWT, wavelet, complex wavelet",
url="https://github.com/fbcotter/pytorch_wavelets",
packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
long_description=read('README.rst'),
classifiers=[
"Development Status :: 3 - Alpha",
"License :: Free To Use But Restricted",
"Programming Language :: Python :: 3",
],
include_package_data=True,
install_requires=['numpy', 'six', 'torch'],
extras_require={
'docs': ['sphinx', 'docutils', 'matplotlib', 'ipython', ],
},
tests_require=['coverage', 'py3nvml', 'dtcwt', 'PyWavelets'],
)
# vim:sw=4:sts=4:et
|