From 3eefbce5b2221d29a0b772ead7ea86b30c6e7218 Mon Sep 17 00:00:00 2001 From: sblunt Date: Fri, 30 Jun 2023 10:14:09 -0700 Subject: [PATCH] pep8 --- setup.py | 60 ++++++++++++++++++++++++++++++++++---------------------- 1 file changed, 37 insertions(+), 23 deletions(-) diff --git a/setup.py b/setup.py index a6237e7e..8c25739f 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ import re USE_C_KEPLER_MODULE = 1 -if ("--disable-cython" in sys.argv): +if "--disable-cython" in sys.argv: sys.argv.remove("--disable-cython") USE_C_KEPLER_MODULE = 0 else: @@ -13,44 +13,58 @@ print("Error: Importing cython build environment failed") USE_C_KEPLER_MODULE = 0 + # auto-updating version code stolen from RadVel def get_property(prop, project): - result = re.search(r'{}\s*=\s*[\'"]([^\'"]*)[\'"]'.format(prop), - open(project + '/__init__.py').read()) + result = re.search( + r'{}\s*=\s*[\'"]([^\'"]*)[\'"]'.format(prop), + open(project + "/__init__.py").read(), + ) return result.group(1) + def get_requires(): reqs = [] - for line in open('requirements.txt', 'r').readlines(): + for line in open("requirements.txt", "r").readlines(): reqs.append(line) return reqs + def get_extensions(): extensions = [] - if(USE_C_KEPLER_MODULE): - extensions = cythonize([Extension("orbitize._kepler", ["orbitize/_kepler.pyx"], include_dirs=[numpy.get_include()])]) + if USE_C_KEPLER_MODULE: + extensions = cythonize( + [ + Extension( + "orbitize._kepler", + ["orbitize/_kepler.pyx"], + include_dirs=[numpy.get_include()], + ) + ] + ) return extensions + setup( - name='orbitize', - version=get_property('__version__', 'orbitize'), - description='orbitize! Turns imaging data into orbits', - url='https://github.com/sblunt/orbitize', - author='', - author_email='', - license='BSD', + name="orbitize", + version=get_property("__version__", "orbitize"), + description="orbitize! Turns imaging data into orbits", + url="https://github.com/sblunt/orbitize", + author="", + author_email="", + license="BSD", packages=find_packages(), - package_data={"":["kernels/*.cu"]}, + package_data={"": ["kernels/*.cu"]}, ext_modules=get_extensions(), include_dirs=[numpy.get_include()], - include_package_data = True, + include_package_data=True, zip_safe=False, classifiers=[ - 'Intended Audience :: Science/Research', - 'Topic :: Scientific/Engineering :: Astronomy', - 'License :: OSI Approved :: BSD License', - 'Programming Language :: Python :: 3.6', - ], - keywords='Orbits Astronomy Astrometry', - install_requires=get_requires() - ) + "Intended Audience :: Science/Research", + "Topic :: Scientific/Engineering :: Astronomy", + "License :: OSI Approved :: BSD License", + "Programming Language :: Python :: 3.6", + ], + keywords="Orbits Astronomy Astrometry", + install_requires=get_requires(), +)