<filename>BOReL/data_management/storage_policy.py
from data_management.multi_task_replay_buffer import MultiTaskReplayBuffer


class MultiTaskPolicyStorage(MultiTaskReplayBuffer):
    def __init__(
        self,
        max_replay_buffer_size,
        obs_dim,
        action_space,
        tasks,
        trajectory_len,
        num_reward_arrays=1,
        **kwargs
    ):
        super().__init__(
            max_replay_buffer_size,
            obs_dim,
            action_space,
            tasks,
            trajectory_len,
            num_reward_arrays,
            **kwargs,
        )

<filename>BOReL/data_management/storage_policy.py
from data_management.multi_task_replay_buffer import MultiTaskReplayBuffer


class MultiTaskPolicyStorage(MultiTaskReplayBuffer):
    def __init__(
        self,
        max_replay_buffer_size,
        obs_dim,
        action_space,
        tasks,
        trajectory_len,
        num_reward_arrays=1,
        **kwargs
    ):
        super().__init__(
            max_replay_buffer_size,
            obs_dim,
            action_space,
            tasks,
            trajectory_len,
            num_reward_arrays,
            **kwargs,
        )
