48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
from __future__ import division
|
|
|
|
from setuptools import setup, find_packages
|
|
|
|
import os
|
|
|
|
|
|
# Minimum dependencies required prior to installation
|
|
INSTALL_REQUIRES = [
|
|
"torch>=2.0.1",
|
|
"torchvision>=0.14.1",
|
|
"tqdm>=4.64.1",
|
|
"tensorboard>=2.11.2",
|
|
"matplotlib>=3.6.2",
|
|
"scipy>=1.9.3",
|
|
"gymnasium>=0.28.1",
|
|
"numpy>=1.23.5",
|
|
"mpld3>=0.5.8",
|
|
"python-dateutil>=2.8.2",
|
|
"Pillow",
|
|
]
|
|
|
|
CLASSIFIERS=[
|
|
"Intended Audience :: Science/Research",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.9",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
]
|
|
|
|
# Installation operation
|
|
setup(
|
|
name="stellar",
|
|
author="Vincent-C",
|
|
version="v2.0.0-pre",
|
|
description="Physics-Embedded Safe Residual RL for Terminal Autonomous Rendezvous",
|
|
keywords=["ppo", "rl", "RPOD", "docking", "safety", "residual-learning"],
|
|
include_package_data=True,
|
|
install_requires=INSTALL_REQUIRES,
|
|
packages=find_packages(include=['stellar', 'stellar.*'], exclude=['misc']),
|
|
classifiers=CLASSIFIERS,
|
|
url="https://github.com/Vince-C156/STELLAR",
|
|
python_requires=">=3.9",
|
|
zip_safe=False,
|
|
)
|