From 07fba3abb04918b4b2fdee1d772cf8892a72d223 Mon Sep 17 00:00:00 2001
From: Shane <shane.morgan@itechpros.com.au>
Date: Fri, 24 May 2024 21:10:06 +1000
Subject: [PATCH] Update BaseEnvironment.py

Addition of action_space_type.
---
 freqtrade/freqai/RL/BaseEnvironment.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/freqtrade/freqai/RL/BaseEnvironment.py b/freqtrade/freqai/RL/BaseEnvironment.py
index ba72c90ed..d1e6650b1 100644
--- a/freqtrade/freqai/RL/BaseEnvironment.py
+++ b/freqtrade/freqai/RL/BaseEnvironment.py
@@ -60,6 +60,7 @@ class BaseEnvironment(gym.Env):
         can_short: bool = False,
         pair: str = "",
         df_raw: DataFrame = DataFrame(),
+        action_space_type: str = "Discrete"
     ):
         """
         Initializes the training/eval environment.
@@ -93,6 +94,7 @@ class BaseEnvironment(gym.Env):
         self.tensorboard_metrics: dict = {}
         self.can_short: bool = can_short
         self.live: bool = live
+        self.action_space_type: str = action_space_type
         if not self.live and self.add_state_info:
             raise OperationalException(
                 "`add_state_info` is not available in backtesting. Change "