forked from AlexanderVNikitin/tsgm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
87 lines (73 loc) · 2.25 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
import os
from setuptools import setup
from setuptools import find_packages
# Function to read version from __version__.py
def get_version():
with open(os.path.join(os.path.dirname(__file__), 'tsgm/version.py')) as f:
exec(f.read())
return locals()['__version__']
name = "tsgm"
version = get_version()
keywords = [
"machine learning",
"deep learning",
"signal processing",
"temporal signal",
"time series",
"generative modeling",
"neural networks",
"GAN",
]
author = "Alexander Nikitin"
url = "https://github.com/AlexanderVNikitin/tsgm"
license = "Apache-2.0"
classifiers = [
"Development Status :: 2 - Pre-Alpha",
"Intended Audience :: Developers",
"Intended Audience :: Financial and Insurance Industry",
"Intended Audience :: Information Technology",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Natural Language :: English",
"Programming Language :: Python :: 3",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Information Analysis",
"Topic :: Scientific/Engineering :: Mathematics",
]
def read_file(filename: str) -> str:
with open(filename, encoding="utf-8") as f:
return f.read().strip()
readme_text = read_file("README.md")
setup(name='tsgm',
version=version,
description='Time Series Generative Modelling Framework',
author=author,
author_email='',
maintainer=author,
maintainer_email='',
url=url,
download_url='',
keywords=keywords,
long_description=readme_text,
long_description_content_type='text/markdown',
license=license,
entry_points={
"console_scripts": ["tsgm-gd=cli.gd:main", "tsgm-eval=cli.eval:main"],
},
install_requires=[
"scipy",
"numpy",
"networkx",
"seaborn",
"scikit-learn",
"prettytable",
"antropy==0.1.6",
"yfinance==0.2.28",
"tqdm",
"dtaidistance >= 2.3.10",
"tensorflow < 2.16",
"tensorflow-probability < 0.24.0",
"statsmodels"
],
package_data={'tsgm': ['README.md']},
packages=find_packages())