the-algorithm/twml/setup.py

30 lines
603 B
Python
Raw Normal View History

2023-04-11 15:04:51 +02:00
import os
from setuptools import find_packages, setup
THIS_DIR = os.path.dirname(os.path.realpath(__file__))
TWML_TEST_DATA_DIR = os.path.join(THIS_DIR, 'twml/tests/data')
data_files = []
for parent, children, files in os.walk(TWML_TEST_DATA_DIR):
data_files += [os.path.join(parent, f) for f in files]
setup(
name='twml',
version='2.0',
description="Tensorflow wrapper for twml",
packages=find_packages(exclude=["build"]),
install_requires=[
'thriftpy2',
'numpy',
'pyyaml',
'future',
'scikit-learn',
'scipy'
],
package_data={
'twml': data_files,
},
)