From 931b93cb63e412961284512270fc82f6ccc8f2e6 Mon Sep 17 00:00:00 2001 From: simonsays1980 Date: Mon, 20 Jan 2025 12:03:36 +0100 Subject: [PATCH] Removed unused imports from replay buffer files. Signed-off-by: simonsays1980 --- rllib/utils/replay_buffers/multi_agent_episode_buffer.py | 1 - rllib/utils/replay_buffers/prioritized_episode_buffer.py | 8 +------- 2 files changed, 1 insertion(+), 8 deletions(-) diff --git a/rllib/utils/replay_buffers/multi_agent_episode_buffer.py b/rllib/utils/replay_buffers/multi_agent_episode_buffer.py index 109894722663e..9b4a0b0f18414 100644 --- a/rllib/utils/replay_buffers/multi_agent_episode_buffer.py +++ b/rllib/utils/replay_buffers/multi_agent_episode_buffer.py @@ -14,7 +14,6 @@ from ray.rllib.utils.annotations import ( DeveloperAPI, override, - OverrideToImplementCustomLogic_CallToSuperRecommended, ) from ray.rllib.utils.spaces.space_utils import batch from ray.rllib.utils.typing import AgentID, ModuleID, SampleBatchType diff --git a/rllib/utils/replay_buffers/prioritized_episode_buffer.py b/rllib/utils/replay_buffers/prioritized_episode_buffer.py index 8cc25728d6175..cb4c5d445dc14 100644 --- a/rllib/utils/replay_buffers/prioritized_episode_buffer.py +++ b/rllib/utils/replay_buffers/prioritized_episode_buffer.py @@ -13,15 +13,9 @@ from ray.rllib.utils import force_list from ray.rllib.utils.annotations import ( override, - OverrideToImplementCustomLogic_CallToSuperRecommended, -) -from ray.rllib.utils.metrics import ( - NUM_AGENT_RESAMPLES, - NUM_MODULE_RESAMPLES, - NUM_RESAMPLES, ) from ray.rllib.utils.replay_buffers.episode_replay_buffer import EpisodeReplayBuffer -from ray.rllib.utils.typing import AgentID, ModuleID, SampleBatchType +from ray.rllib.utils.typing import ModuleID, SampleBatchType class PrioritizedEpisodeReplayBuffer(EpisodeReplayBuffer):