Apply isort

Lightning-Universe · Dec 17, 2020 · 6c7d572 · 6c7d572
1 parent b394889
commit 6c7d572
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pl_bolts/models/rl/common/gym_wrappers.py b/pl_bolts/models/rl/common/gym_wrappers.py
@@ -7,13 +7,13 @@
 import numpy as np
 import torch
 
-from pl_bolts.utils import _OPENCV_AVAILABLE, _GYM_AVAILABLE
+from pl_bolts.utils import _GYM_AVAILABLE, _OPENCV_AVAILABLE
 from pl_bolts.utils.warnings import warn_missing_pkg
 
 if _GYM_AVAILABLE:
-    from gym import Wrapper, ObservationWrapper
-    from gym import make as gym_make
     import gym.spaces
+    from gym import ObservationWrapper, Wrapper
+    from gym import make as gym_make
 else:  # pragma: no-cover
     warn_missing_pkg('gym')
     Wrapper = object

diff --git a/pl_bolts/models/rl/dqn_model.py b/pl_bolts/models/rl/dqn_model.py
@@ -18,11 +18,11 @@
 from pl_bolts.datamodules.experience_source import Experience, ExperienceSourceDataset
 from pl_bolts.losses.rl import dqn_loss
 from pl_bolts.models.rl.common.agents import ValueAgent
+from pl_bolts.models.rl.common.gym_wrappers import make_environment
 from pl_bolts.models.rl.common.memory import MultiStepBuffer
 from pl_bolts.models.rl.common.networks import CNN
 from pl_bolts.utils import _GYM_AVAILABLE
 from pl_bolts.utils.warnings import warn_missing_pkg
-from pl_bolts.models.rl.common.gym_wrappers import make_environment
 
 if _GYM_AVAILABLE:
     from gym import Env