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