removing datasets + imports optimization

rl_coach/exploration_policies/additive_noise.py

@@ -17,11 +17,11 @@
 from typing import List
 
 import numpy as np
-from rl_coach.schedules import Schedule, LinearSchedule
-from rl_coach.spaces import ActionSpace, BoxActionSpace
 
 from rl_coach.core_types import RunPhase, ActionType
 from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy, ExplorationParameters
+from rl_coach.schedules import Schedule, LinearSchedule
+from rl_coach.spaces import ActionSpace, BoxActionSpace
 
 
 # TODO: consider renaming to gaussian sampling

rl_coach/exploration_policies/boltzmann.py

@@ -17,11 +17,11 @@
 from typing import List
 
 import numpy as np
-from rl_coach.schedules import Schedule
-from rl_coach.spaces import ActionSpace
 
 from rl_coach.core_types import RunPhase, ActionType
 from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy, ExplorationParameters
+from rl_coach.schedules import Schedule
+from rl_coach.spaces import ActionSpace
 
 
 class BoltzmannParameters(ExplorationParameters):

rl_coach/exploration_policies/bootstrapped.py

@@ -17,13 +17,13 @@
 from typing import List
 
 import numpy as np
-from rl_coach.exploration_policies.additive_noise import AdditiveNoiseParameters
-from rl_coach.exploration_policies.e_greedy import EGreedy, EGreedyParameters
-from rl_coach.schedules import Schedule, LinearSchedule
-from rl_coach.spaces import ActionSpace
 
 from rl_coach.core_types import RunPhase, ActionType
+from rl_coach.exploration_policies.additive_noise import AdditiveNoiseParameters
+from rl_coach.exploration_policies.e_greedy import EGreedy, EGreedyParameters
 from rl_coach.exploration_policies.exploration_policy import ExplorationParameters
+from rl_coach.schedules import Schedule, LinearSchedule
+from rl_coach.spaces import ActionSpace
 
 
 class BootstrappedParameters(EGreedyParameters):

rl_coach/exploration_policies/categorical.py

@@ -17,10 +17,10 @@
 from typing import List
 
 import numpy as np
-from rl_coach.spaces import ActionSpace
 
 from rl_coach.core_types import RunPhase, ActionType
 from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy, ExplorationParameters
+from rl_coach.spaces import ActionSpace
 
 
 class CategoricalParameters(ExplorationParameters):

rl_coach/exploration_policies/e_greedy.py

@@ -17,15 +17,15 @@
 from typing import List
 
 import numpy as np
+
+from rl_coach.core_types import RunPhase, ActionType
 from rl_coach.exploration_policies.additive_noise import AdditiveNoiseParameters
+from rl_coach.exploration_policies.exploration_policy import ExplorationParameters
+from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy
 from rl_coach.schedules import Schedule, LinearSchedule
 from rl_coach.spaces import ActionSpace, DiscreteActionSpace, BoxActionSpace
 from rl_coach.utils import dynamic_import_and_instantiate_module_from_params
 
-from rl_coach.core_types import RunPhase, ActionType
-from rl_coach.exploration_policies.exploration_policy import ExplorationParameters
-from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy
-
 
 class EGreedyParameters(ExplorationParameters):
     def __init__(self):

rl_coach/exploration_policies/exploration_policy.py

@@ -17,9 +17,8 @@
 from typing import List
 
 from rl_coach.base_parameters import Parameters
-from rl_coach.spaces import ActionSpace
-
 from rl_coach.core_types import RunPhase, ActionType
+from rl_coach.spaces import ActionSpace
 
 
 class ExplorationParameters(Parameters):

rl_coach/exploration_policies/greedy.py

@@ -17,10 +17,10 @@
 from typing import List
 
 import numpy as np
-from rl_coach.spaces import ActionSpace, DiscreteActionSpace, BoxActionSpace
 
 from rl_coach.core_types import ActionType
 from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy, ExplorationParameters
+from rl_coach.spaces import ActionSpace, DiscreteActionSpace, BoxActionSpace
 
 
 class GreedyParameters(ExplorationParameters):

rl_coach/exploration_policies/ou_process.py

@@ -17,10 +17,10 @@
 from typing import List
 
 import numpy as np
-from rl_coach.spaces import ActionSpace, BoxActionSpace, GoalsSpace
 
 from rl_coach.core_types import RunPhase, ActionType
 from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy, ExplorationParameters
+from rl_coach.spaces import ActionSpace, BoxActionSpace, GoalsSpace
 
 
 # Based on on the description in:

rl_coach/exploration_policies/truncated_normal.py

@@ -17,12 +17,12 @@
 from typing import List
 
 import numpy as np
-from rl_coach.schedules import Schedule, LinearSchedule
 from scipy.stats import truncnorm
-from rl_coach.spaces import ActionSpace, BoxActionSpace
 
 from rl_coach.core_types import RunPhase, ActionType
 from rl_coach.exploration_policies.exploration_policy import ExplorationPolicy, ExplorationParameters
+from rl_coach.schedules import Schedule, LinearSchedule
+from rl_coach.spaces import ActionSpace, BoxActionSpace
 
 
 class TruncatedNormalParameters(ExplorationParameters):

rl_coach/exploration_policies/ucb.py

@@ -17,13 +17,13 @@
 from typing import List
 
 import numpy as np
-from rl_coach.exploration_policies.additive_noise import AdditiveNoiseParameters
-from rl_coach.exploration_policies.e_greedy import EGreedy, EGreedyParameters
-from rl_coach.schedules import Schedule, LinearSchedule, PieceWiseSchedule
-from rl_coach.spaces import ActionSpace
 
 from rl_coach.core_types import RunPhase, ActionType, EnvironmentSteps
+from rl_coach.exploration_policies.additive_noise import AdditiveNoiseParameters
+from rl_coach.exploration_policies.e_greedy import EGreedy, EGreedyParameters
 from rl_coach.exploration_policies.exploration_policy import ExplorationParameters
+from rl_coach.schedules import Schedule, LinearSchedule, PieceWiseSchedule
+from rl_coach.spaces import ActionSpace
 
 
 class UCBParameters(EGreedyParameters):