lets create trading
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user