|
- import os
- from setuptools import setup, find_packages
- import subprocess
- import logging
-
- PACKAGE_NAME = 'tods'
-
- def read_file_entry_points(fname):
- with open(fname) as entry_points:
- return entry_points.read()
-
- def merge_entry_points():
- entry_list = ['entry_points.ini', 'entry_points_common.ini']
- merge_entry = []
- for entry_name in entry_list:
- entry_point = read_file_entry_points(entry_name).replace(' ', '')
- path_list = entry_point.split('\n')[1:]
- merge_entry += path_list
- entry_point_merge = dict()
- entry_point_merge['d3m.primitives'] = list(set(merge_entry)) # remove dumplicated elements
- return entry_point_merge
-
- setup(
- name=PACKAGE_NAME,
- version='0.0.1',
- description='Primitives for time-series outlier detection',
- author='DATA Lab',
- packages=find_packages(exclude=['contrib', 'docs', 'tests*']),
- install_requires=[
- 'd3m',
- 'Jinja2',
- 'GitPython==3.1.0',
- 'simplejson==3.12.0',
- 'scikit-learn==0.22.0',
- 'statsmodels==0.11.1',
- 'PyWavelets>=1.1.1',
- 'tensorflow', # should be removed later
- 'keras', # should be removed later
- 'pyod',
- 'nimfa==1.4.0',
- 'stumpy==1.4.0',
- 'more-itertools==8.5.0',
- 'gitdb2==2.0.6',
- 'gitdb==0.6.4'
- ],
-
- entry_points = merge_entry_points()
-
- )
-
- try:
- subprocess.run(['pip', 'install', '-r', 'requirements.txt'])
- except Exception as e:
- print(e)
|