lets create trading

This commit is contained in:
2023-01-14 11:54:14 +01:00
parent 058eea02b2
commit 8801ebaf58
4 changed files with 24 additions and 14 deletions

View File

@@ -82,16 +82,18 @@ class SB3EconConverter(VecEnv, gym.Env):
def seed(self, seed: Optional[int] = None) -> List[Union[None, int]]:
if seed is None:
seed = np.random.randint(0, 2**32 - 1)
seeds = []
for idx, env in enumerate(self.envs):
seeds.append(env.seed(seed + idx))
self.econ.seed(seed)
seeds=[seed]
return seeds
def action_masks(self):
"""Returns action masks for agents and current obs"""
masks=[]
for obs in self.curr_obs:
masks.append(self.curr_obs[obs]["action_mask"])
mask=[]
for num in self.curr_obs[obs]["action_mask"]:
mask.append(num==1.0)
masks.append(mask)
return masks
def close(self) -> None: