def __init__(self, **params):
from catalyst.contrib.registry import Registry
config_ = params.copy()
self.strategies: List[ExplorationStrategy] = []
self.probs = []
for key, expl in config_.items():
After Change
for params_ in params:
exploration_name = params_.pop("exploration")
probability = params_.pop("probability")
strategy_fn = EXPLORATION.get(exploration_name)
if issubclass(strategy_fn, EpsilonGreedy):
assert isinstance(env.action_space, Discrete)