From ed4ae2e01262668ce4ef698aa0796968696a74c8 Mon Sep 17 00:00:00 2001 From: Prabhat Date: Fri, 6 Nov 2020 01:11:28 +0900 Subject: [PATCH] Applies isort --- examples/her/train_dqn_bit_flip.py | 9 ++------- pfrl/replay_buffers/__init__.py | 6 +++--- pfrl/replay_buffers/hindsight.py | 2 +- 3 files changed, 6 insertions(+), 11 deletions(-) diff --git a/examples/her/train_dqn_bit_flip.py b/examples/her/train_dqn_bit_flip.py index 0354bb344..a430a1bef 100644 --- 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 index 3319041f8..d33f1bfba 100644 --- 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 index fa24cdffd..92b8f1eb2 100644 --- 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):