mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-11-13 07:05:10 +01:00
29 lines
649 B
Python
29 lines
649 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,
|
|
},
|
|
)
|