text: #!/usr/bin/env python
  # -*- coding: utf-8 -*-
  
  """The setup script."""
  
  from setuptools import find_packages, setup
  
  with open('README.md', encoding='utf-8') as readme_file:
      readme = readme_file.read()
  
  with open('HISTORY.md', encoding='utf-8') as history_file:
      history = history_file.read()
  
  install_requires = [
      'packaging',
      'numpy>=1.18.0,<2',
      'pandas>=1.1,<1.1.5',
      'scikit-learn>=0.23,<1',
      'torch>=1.4,<2',
      'torchvision>=0.5.0,<1',
      'rdt>=0.5.0,<0.6',
  ]
  
  setup_requires = [
      'pytest-runner>=2.11.1',
  ]
  
  tests_require = [
      'pytest>=3.4.2',
      'pytest-cov>=2.6.0',
  ]
  
  development_requires = [
      # general
      'pip>=9.0.1',
      'bumpversion>=0.5.3,<0.6',
      'watchdog>=0.8.3,<0.11',
  
      # style check
      'flake8>=3.7.7,<4',
      'isort>=4.3.4,<5',
  
      # fix style issues
      'autoflake>=1.1,<2',
      'autopep8>=1.4.3,<1.6',
  
      # distribute on PyPI
      'twine>=1.10.0,<4',
      'wheel>=0.30.0',
  
      # Advanced testing
      'coverage>=4.5.1,<6',
      'tox>=2.9.1,<4',
  
      'invoke',
  ]
  
  setup(
      author='MIT Data To AI Lab',
      author_email='dailabmit@gmail.com',
      classifiers=[
          'Development Status :: 2 - Pre-Alpha',
          'Intended Audience :: Developers',
          'License :: OSI Approved :: MIT License',
          'Natural Language :: English',
          'Programming Language :: Python :: 3',
          'Programming Language :: Python :: 3.6',
          'Programming Language :: Python :: 3.7',
          'Programming Language :: Python :: 3.8',
      ],
      description='Conditional GAN for Tabular Data',
      entry_points={
          'console_scripts': [
              'ctgan=ctgan.__main__:main'
          ],
      },
      extras_require={
          'test': tests_require,
          'dev': development_requires + tests_require,
      },
      install_package_data=True,
      install_requires=install_requires,
      license='MIT license',
      long_description=readme + '\n\n' + history,
      long_description_content_type='text/markdown',
      include_package_data=True,
      keywords='ctgan CTGAN',
      name='ctgan',
      packages=find_packages(include=['ctgan', 'ctgan.*']),
      python_requires='>=3.6,<3.9',
      setup_requires=setup_requires,
      test_suite='tests',
      tests_require=tests_require,
      url='https://github.com/sdv-dev/CTGAN',
      version='0.4.4.dev1',
      zip_safe=False,
  )
  