setup.py

"""
A package to train and infer DDPG Tensorflow 2
"""

from pathlib import Path
import setuptools
from setuptools import find_packages

PACKAGE_NAME: str = "ddpg-tf2"

# read the contents of your README file
this_directory = Path(__file__).parent
long_description = (this_directory / "README.md").read_text()

REQUIRED = [
  "tensorflow-gpu==2.7.2",
  "tqdm==4.37.0",
  "gym==0.17.2",
  "box2d-py==2.3.8",
  "matplotlib==3.1.1",
  "pytest==7.1.2",
  "bandit==1.7.4",
  "pylint==2.14.4"
]

setuptools.setup(
  name=PACKAGE_NAME,
  version='1.0.1',
  description="""
    Simple yet effective DDPG implementation for continuous action space,
    written in Tensorflow 2
  """,
  long_description=long_description,
  long_description_content_type='text/markdown',
  author="Samuel Matthew Koesnadi",
  author_email="samuelmat19@gmail.com",
  install_requires=REQUIRED,
  packages=find_packages(),
  entry_points = {
    'console_scripts': [f'{PACKAGE_NAME}=src.command_line:main'],
  }
)