code: def __init__(
      self,
      capacity: Optional[int] = None,
      wait_priority_after_sampling: bool = True,
      initial_max_priority: float = 1.0,
  ):
      self.capacity = capacity
      self.data: Deque = collections.deque()
      self.priority_sums = SumTreeQueue()
      self.priority_mins = MinTreeQueue()
      self.max_priority = initial_max_priority
      self.wait_priority_after_sampling = wait_priority_after_sampling
      self.flag_wait_priority = False
  
imports: import collections
  from typing import Any, Callable, Deque, Generic, List, Optional, Sequence, Tuple, TypeVar
  import numpy
  from pfrl.utils.random import sample_n_k
label_desc: Use image node_id 1 to create a new PrioritizedBuffer object from inherited base classes: Generic with example: obj = PrioritizedBuffer(capacity, wait_priority_after_sampling, initial_max_priority)