twitter-algorithm/twml/setup.py
twitter-team ef4c5eb65e Twitter Recommendation Algorithm
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
2023-03-31 17:36:31 -05:00

30 lines
603 B
Python

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,
},
)