import argparse

from setuptools import setup
from setuptools import find_packages

# parse args
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', '-g', action='store_true')
args, unknown = parser.parse_known_args()

tf_suffix = '-gpu' if args.gpu else ''

setup(name='forkan',
      version='0.0.1',
      description='Deep Learning experiments',
      author='<NAME>',
      author_email='<EMAIL>',
      url='https://github.com/llach/dl-toolkit',
      install_requires=['tensorflow{}>=1.11.0'.format(tf_suffix),
                        'numpy>=1.15.3',
                        'tabulate>=0.8.2',
                        'Pillow>=5.3.0',
                        'gym>=0.10.9',
                        'scikit-image>=0.14.2',
                        'coloredlogs>=10.0',
                        'matplotlib>=3.0.0',
                        'keras>=2.2.4',
                        'tqdm>=4.30.0',
                        'baselines'],
      packages=find_packages())

import argparse

from setuptools import setup
from setuptools import find_packages

# parse args
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', '-g', action='store_true')
args, unknown = parser.parse_known_args()

tf_suffix = '-gpu' if args.gpu else ''

setup(name='forkan',
      version='0.0.1',
      description='Deep Learning experiments',
      author='<NAME>',
      author_email='<EMAIL>',
      url='https://github.com/llach/dl-toolkit',
      install_requires=['tensorflow{}>=1.11.0'.format(tf_suffix),
                        'numpy>=1.15.3',
                        'tabulate>=0.8.2',
                        'Pillow>=5.3.0',
                        'gym>=0.10.9',
                        'scikit-image>=0.14.2',
                        'coloredlogs>=10.0',
                        'matplotlib>=3.0.0',
                        'keras>=2.2.4',
                        'tqdm>=4.30.0',
                        'baselines'],
      packages=find_packages())

# parse args