alphalens/setup.py at master · cloudQuant/alphalens · GitHub

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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/usr/bin/env python
from setuptools import setup, find_packages
import sys
import os

# Simple version management to avoid dependency issues during CI
def get_version():
    """Get version from _version.py or fallback to default"""
    try:
        # Try to read version from _version.py
        version_file = os.path.join(os.path.dirname(__file__), 'alphalens', '_version.py')
        if os.path.exists(version_file):
            with open(version_file, 'r') as f:
                content = f.read()
                # Extract version from the file
                for line in content.split('\n'):
                    if line.strip().startswith('version ='):
                        return line.split('=')[1].strip().strip('"\'')
    except:
        pass
    return "1.0.0+dev"

def get_cmdclass():
    """Return empty cmdclass to avoid versioneer dependency"""
    return {}

long_description = ''

if 'upload' in sys.argv:
    with open('README.rst') as f:
        long_description = f.read()

install_reqs = [
    'matplotlib',
    'numpy',
    'pandas',
    'scipy',
    'seaborn',
    'statsmodels',
    'IPython',
    # Note: empyrical should be installed via git
    # pip install -U git+https://github.com/cloudQuant/empyrical.git
    'empyrical @ git+https://github.com/cloudQuant/empyrical.git',
]

extra_reqs = {
    'test': [
        "pytest>=6.0",
        "pytest-xdist>=2.0",
        "pytest-cov>=2.10",
        "pytest-sugar>=0.9",
        "pytest-benchmark>=3.4",
        "pytest-picked>=0.4",
        "parameterized>=0.7",
        "flake8>=3.7.9",
    ],
}

if __name__ == "__main__":
    setup(
        name='alphalens',
        version=get_version(),
        cmdclass=get_cmdclass(),
        description='Performance analysis of predictive (alpha) stock factors',
        author='Quantopian Inc. and cloudQuant',
        author_email='yunjinqi@gmail.com',
        packages=find_packages(include='alphalens.*'),
        package_data={
            'alphalens': ['examples/*'],
        },
        long_description=long_description,
        classifiers=[
            'Development Status :: 5 - Production/Stable',
            'Intended Audience :: Developers',
            'License :: OSI Approved :: Apache Software License',
            'Natural Language :: English',
            'Operating System :: OS Independent',
            'Programming Language :: Python :: 2.7',
            'Programming Language :: Python :: 3.4',
            'Programming Language :: Python :: 3.5',
            'Programming Language :: Python',
            'Topic :: Utilities',
            'Topic :: Office/Business :: Financial',
            'Topic :: Scientific/Engineering :: Information Analysis',
        ],
        url='https://gitee.com/yunjinqi/alphalens',
        install_requires=install_reqs,
        extras_require=extra_reqs,
    )