text: import numpy as np
  from data_management.multi_task_replay_buffer import MultiTaskReplayBuffer
  
  
  class MultiTaskVAEStorage(MultiTaskReplayBuffer):
      def __init__(
          self,
          max_replay_buffer_size,
          obs_dim,
          action_space,
          tasks,
          trajectory_len,
      ):
          super().__init__(
              max_replay_buffer_size, obs_dim, action_space, tasks, trajectory_len
          )
  