From 3fa8beb7ae10aa277359e1cbc746b1ac253d456e Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Mon, 1 Apr 2024 00:17:50 +0200 Subject: [PATCH] Make pca optional dependency --- setup.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 8bea528..c28212d 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,15 @@ setup( url='https://git.dominik-roth.eu/dodox/metastable-baselines2', author='Dominik Roth', author_email='mail@dominik-roth.eu', - # description='Description of my package', - packages=['.'], - install_requires=['gymnasium', 'stable_baselines3==2.1.0', '-e git+ssh://git@dominik-roth.eu/dodox/PriorConditionedAnnealing.git#egg=priorConditionedAnnealing'], + packages=find_packages(), + install_requires=[ + 'gymnasium', + 'stable_baselines3==2.1.0', + ], + extras_require={ + 'pca': [ + 'priorConditionedAnnealing @ git+ssh://git@dominik-roth.eu/dodox/PriorConditionedAnnealing.git#egg=priorConditionedAnnealing' + ], + }, ) +