forked from LineaLabs/lineapy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
115 lines (103 loc) · 2.86 KB
/
setup.py
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import io
import os
import re
# read the contents of your README file
from pathlib import Path
from setuptools import find_packages, setup
this_directory = Path(__file__).parent
LONG_DESCRIPTION = (this_directory / "README.md").read_text()
# LONG_DESCRIPTION = """"""
DESCRIPTION = ""
NAME = "lineapy"
AUTHOR = "linealabs"
AUTHOR_EMAIL = "dev@linea.ai"
URL = "linea.ai"
def read(path, encoding="utf-8"):
path = os.path.join(os.path.dirname(__file__), path)
with io.open(path, encoding=encoding) as fp:
return fp.read()
def version(path):
"""Obtain the package version from a python file e.g. pkg/__init__.py
See <https://packaging.python.org/en/latest/single_source_version.html>.
"""
version_file = read(path)
version_match = re.search(
r"""^__version__ = ['"]([^'"]*)['"]""", version_file, re.M
)
if version_match:
return version_match.group(1)
raise RuntimeError("Unable to find version string.")
DOWNLOAD_URL = "linea.ai"
LICENSE = "TODO"
VERSION = version("lineapy/__init__.py")
setup(
name=NAME,
version=VERSION,
description=DESCRIPTION,
long_description=LONG_DESCRIPTION,
long_description_content_type="text/markdown",
author=AUTHOR,
author_email=AUTHOR_EMAIL,
url=URL,
download_url=DOWNLOAD_URL,
license=LICENSE,
classifiers=["Development Status :: 2 - Pre-Alpha"],
packages=find_packages(),
# https://python-packaging.readthedocs.io/en/latest/command-line-scripts.html#the-console-scripts-entry-point
entry_points={"console_scripts": ["lineapy=lineapy.cli.cli:linea_cli"]},
install_requires=[
"Pillow",
"astor",
"click>=8.0.0",
"pydantic",
"SQLAlchemy",
"networkx",
"black",
"rich",
"astpretty",
"scour==0.38.2",
"pyyaml",
"asttokens",
"isort",
"graphviz",
"IPython",
"jinja2",
"nbformat",
"nbconvert",
],
extras_require={
"dev": [
"altair",
"pandas",
"sklearn",
"flake8",
"fastparquet",
"syrupy==1.4.5",
"mypy",
"pytest",
"matplotlib",
# Coveralls doesn't work with 6.0
# https://github.com/TheKevJames/coveralls-python/issues/326
"coverage[toml]<6.0",
"pytest-cov",
"jupyterlab",
"pdbpp",
"pytest-virtualenv",
"nbval",
"coveralls",
"seaborn",
"pre-commit",
"SQLAlchemy[mypy]>=1.4.0",
"sphinx",
"sphinx_rtd_theme",
"types-PyYAML",
"pg",
"psycopg2",
"pytest-xdist",
"sphinx-autobuild",
# For benchmark CI
"scipy",
]
},
include_package_data=True,
)