Applies isort

pfnet · Nov 5, 2020 · ed4ae2e · ed4ae2e
1 parent 573c7a2
commit ed4ae2e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 11 deletions.
diff --git a/examples/her/train_dqn_bit_flip.py b/examples/her/train_dqn_bit_flip.py
@@ -6,14 +6,9 @@
 import torch
 import torch.nn as nn
 
-from pfrl.q_functions import DiscreteActionValueHead
-from pfrl import agents
-from pfrl import experiments
-from pfrl import explorers
-from pfrl import utils
-from pfrl import replay_buffers
-
+from pfrl import agents, experiments, explorers, replay_buffers, utils
 from pfrl.initializers import init_chainer_default
+from pfrl.q_functions import DiscreteActionValueHead
 
 
 def reward_fn(dg, ag):

diff --git a/pfrl/replay_buffers/__init__.py b/pfrl/replay_buffers/__init__.py
@@ -1,6 +1,8 @@
 from pfrl.replay_buffers.episodic import EpisodicReplayBuffer  # NOQA
-from pfrl.replay_buffers.hindsight import HindsightReplayStrategy  # NOQA
 from pfrl.replay_buffers.hindsight import HindsightReplayBuffer  # NOQA
+from pfrl.replay_buffers.hindsight import HindsightReplayStrategy  # NOQA
+from pfrl.replay_buffers.hindsight import ReplayFinalGoal  # NOQA
+from pfrl.replay_buffers.hindsight import ReplayFutureGoal  # NOQA
 from pfrl.replay_buffers.persistent import PersistentEpisodicReplayBuffer  # NOQA
 from pfrl.replay_buffers.persistent import PersistentReplayBuffer  # NOQA
 from pfrl.replay_buffers.prioritized import PrioritizedReplayBuffer  # NOQA
@@ -9,5 +11,3 @@
     PrioritizedEpisodicReplayBuffer,
 )
 from pfrl.replay_buffers.replay_buffer import ReplayBuffer  # NOQA
-from pfrl.replay_buffers.hindsight import ReplayFinalGoal  # NOQA
-from pfrl.replay_buffers.hindsight import ReplayFutureGoal  # NOQA
diff --git a/pfrl/replay_buffers/hindsight.py b/pfrl/replay_buffers/hindsight.py
@@ -2,8 +2,8 @@
 
 import numpy as np
 
-from pfrl.replay_buffers.episodic import EpisodicReplayBuffer
 from pfrl.replay_buffer import random_subseq
+from pfrl.replay_buffers.episodic import EpisodicReplayBuffer
 
 
 def relabel_transition_goal(transition, goal_transition, reward_fn, swap_keys_list):