the-algorithm/twml/setup.py

29 lines
649 B
Python
Raw Normal View History

import os
from setuptools import find_packages, setup
THIS_DIR = os.path.dirname(os.path.realpath(__file__))
2023-04-17 06:19:03 +02:00
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):
2023-04-17 06:19:03 +02:00
data_files += [os.path.join(parent, f) for f in files]
setup(
2023-04-17 06:19:03 +02:00
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,
},
)