From afd54d39a5b675e0ce19afcb2bfbc9c2465a9d16 Mon Sep 17 00:00:00 2001
From: steam <steam@system-root.ru>
Date: Sun, 11 Jun 2023 20:00:12 +0300
Subject: [PATCH] add action_masks

---
 freqtrade/freqai/RL/BaseEnvironment.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/freqtrade/freqai/RL/BaseEnvironment.py b/freqtrade/freqai/RL/BaseEnvironment.py
index 42e644f0a..d1a399c48 100644
--- a/freqtrade/freqai/RL/BaseEnvironment.py
+++ b/freqtrade/freqai/RL/BaseEnvironment.py
@@ -141,6 +141,9 @@ class BaseEnvironment(gym.Env):
         Unique to the environment action count. Must be inherited.
         """
 
+    def action_masks(self) -> list[bool]:
+        return [self._is_valid(action.value) for action in self.actions]
+
     def seed(self, seed: int = 1):
         self.np_random, seed = seeding.np_random(seed)
         return [seed]