fix action_masks typing list

This commit is contained in:
steam
2023-06-11 21:44:57 +03:00
parent c36547a563
commit 5dee86eda7

View File

@@ -2,7 +2,7 @@ import logging
import random import random
from abc import abstractmethod from abc import abstractmethod
from enum import Enum from enum import Enum
from typing import Optional, Type, Union from typing import List, Optional, Type, Union
import gymnasium as gym import gymnasium as gym
import numpy as np import numpy as np
@@ -141,7 +141,7 @@ class BaseEnvironment(gym.Env):
Unique to the environment action count. Must be inherited. Unique to the environment action count. Must be inherited.
""" """
def action_masks(self) -> list[bool]: def action_masks(self) -> List[bool]:
return [self._is_valid(action.value) for action in self.actions] return [self._is_valid(action.value) for action in self.actions]
def seed(self, seed: int = 1): def seed(self, seed: int = 1):