Improve setup script

freqtrade · Dec 7, 2023 · 8689796 · 8689796
1 parent aba74d4
commit 8689796
Showing 1 changed file with 57 additions and 37 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,5 @@
-try:
-    from setuptools import setup
-except ImportError:
-    from distutils.core import setup
+from setuptools import find_packages, setup
+
 try:
     import builtins
 except ImportError:
@@ -17,36 +15,58 @@
 
 VERSION = skopt.__version__
 
-CLASSIFIERS = ['Intended Audience :: Science/Research',
-               'Intended Audience :: Developers',
-               'License :: OSI Approved :: BSD License',
-               'Programming Language :: Python',
-               'Topic :: Software Development',
-               'Topic :: Scientific/Engineering',
-               'Operating System :: Microsoft :: Windows',
-               'Operating System :: POSIX',
-               'Operating System :: Unix',
-               'Operating System :: MacOS',
-               'Programming Language :: Python :: 3.6',
-               'Programming Language :: Python :: 3.7',
-               'Programming Language :: Python :: 3.8']
-
-
-setup(name='scikit-optimize',
-      version=VERSION,
-      description='Sequential model-based optimization toolbox.',
-      long_description=open('README.rst').read(),
-      url='https://scikit-optimize.github.io/',
-      license='BSD 3-clause',
-      author='The scikit-optimize contributors',
-      classifiers=CLASSIFIERS,
-      packages=['skopt', 'skopt.learning', 'skopt.optimizer', 'skopt.space',
-                'skopt.learning.gaussian_process', 'skopt.sampler'],
-      install_requires=['joblib>=0.11', 'pyaml>=16.9', 'numpy>=1.13.3',
-                        'scipy>=0.19.1',
-                        'scikit-learn>=0.20.0'],
-      extras_require={
-        'plots':  ["matplotlib>=2.0.0"]
-        }
-
-      )
+setup(
+    name='scikit-optimize',
+    version=VERSION,
+    description='Sequential model-based optimization toolbox.',
+    long_description=open('README.rst').read(),
+    url='https://scikit-optimize.github.io/',
+    license='BSD 3-clause',
+    author='The scikit-optimize contributors',
+    packages=find_packages(include=('skopt.*',),
+                           exclude=('*.tests',)),
+    # use_scm_version=True,
+    python_requires='>= 3.8',
+    setup_requires=[
+        'setuptools_scm',
+    ],
+    install_requires=[
+        'joblib>=1.2',
+        'pyaml>=16.9',
+        'numpy>=1.17',
+        'scipy>=1.5',
+        'scikit-learn>1.1',
+    ],
+    extras_require={
+        'plots': [
+            "matplotlib>=2.0.0",
+        ],
+        'dev': [
+            'flake8',
+            'pytest',
+            'pytest-cov',
+            'pytest-xdist',
+        ],
+        'doc': [
+            'sphinx',
+            'sphinx-gallery>=0.6',
+            'memory_profiler',
+            'numpydoc',
+        ],
+    },
+    classifiers=[
+        'Development Status :: 5 - Production/Stable',
+        'Intended Audience :: Science/Research',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: BSD License',
+        'Programming Language :: Python :: 3',
+        'Topic :: Scientific/Engineering :: Artificial Intelligence',
+        'Topic :: Scientific/Engineering :: Visualization',
+        'Topic :: Software Development :: Libraries :: Python Modules',
+        'Operating System :: MacOS',
+        'Operating System :: Microsoft :: Windows',
+        'Operating System :: OS Independent',
+        'Operating System :: POSIX',
+        'Operating System :: Unix',
+    ],
+    )