def __init__()

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

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

    def __init__(self,
                 model: ModelLike,
                 eps: float = 0.1,
                 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:
        super().__init__()

        self.model = model
        self.eps = eps

        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

        self.trajectory = []