Removed unused imports from replay buffer files.

Signed-off-by: simonsays1980 <simon.zehnder@gmail.com>
ray-project · Jan 20, 2025 · 931b93c · 931b93c
1 parent aac1507
commit 931b93c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git 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
@@ -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):