def __init__()

in rlmeta/agents/dqn/apex_dqn_agent.py [0:0]

• 23 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self,
                 model: ModelLike,
                 eps_func: Callable[[int], float],
                 replay_buffer: Optional[ReplayBufferLike] = None,
                 controller: Optional[ControllerLike] = None,
                 optimizer: Optional[torch.optim.Optimizer] = None,
                 batch_size: int = 128,
                 grad_clip: float = 50.0,
                 multi_step: int = 1,
                 gamma: float = 0.99,
                 sync_every_n_steps: int = 10,
                 push_every_n_steps: int = 1) -> None:
        self._model = model
        self._eps_func = eps_func
        self._replay_buffer = replay_buffer
        self._controller = controller
        self._optimizer = optimizer
        self._batch_size = batch_size
        self._grad_clip = grad_clip
        self._multi_step = multi_step
        self._gamma = gamma
        self._sync_every_n_steps = sync_every_n_steps
        self._push_every_n_steps = push_every_n_steps