add sb3_contrib models to the available agents. include sb3_contrib in requirements.

This commit is contained in:
robcaulk
2022-08-21 19:58:36 +02:00
parent 8b3a8234ac
commit d88a0dbf82
3 changed files with 35 additions and 25 deletions

View File

@@ -223,12 +223,10 @@ class Base5ActionRLEnv(gym.Env):
(action == Actions.Neutral.value and self._position == Positions.Long) or
(action == Actions.Short_enter.value and self._position == Positions.Short) or
(action == Actions.Short_enter.value and self._position == Positions.Long) or
# (action == Actions.Short_exit.value and self._position == Positions.Short) or
(action == Actions.Short_exit.value and self._position == Positions.Long) or
(action == Actions.Short_exit.value and self._position == Positions.Neutral) or
(action == Actions.Long_enter.value and self._position == Positions.Long) or
(action == Actions.Long_enter.value and self._position == Positions.Short) or
# (action == Actions.Long_exit.value and self._position == Positions.Long) or
(action == Actions.Long_exit.value and self._position == Positions.Short) or
(action == Actions.Long_exit.value and self._position == Positions.Neutral))