removing datasets + imports optimization

rl_coach/filters/action/action_filter.py

@@ -14,10 +14,9 @@
 # limitations under the License.
 #
 
-from rl_coach.spaces import ActionSpace
-
 from rl_coach.core_types import ActionType
 from rl_coach.filters.filter import Filter
+from rl_coach.spaces import ActionSpace
 
 
 class ActionFilter(Filter):

rl_coach/filters/action/attention_discretization.py

@@ -17,8 +17,8 @@
 from typing import Union, List
 
 import numpy as np
-from rl_coach.filters.action.box_discretization import BoxDiscretization
 
+from rl_coach.filters.action.box_discretization import BoxDiscretization
 from rl_coach.filters.action.partial_discrete_action_space_map import PartialDiscreteActionSpaceMap
 from rl_coach.spaces import AttentionActionSpace, BoxActionSpace, DiscreteActionSpace
 

rl_coach/filters/action/box_masking.py

@@ -17,10 +17,10 @@
 from typing import Union
 
 import numpy as np
-from rl_coach.spaces import BoxActionSpace
 
 from rl_coach.core_types import ActionType
 from rl_coach.filters.action.action_filter import ActionFilter
+from rl_coach.spaces import BoxActionSpace
 
 
 class BoxMasking(ActionFilter):

rl_coach/filters/action/linear_box_to_box_map.py

@@ -17,10 +17,10 @@
 from typing import Union
 
 import numpy as np
-from rl_coach.spaces import BoxActionSpace
 
 from rl_coach.core_types import ActionType
 from rl_coach.filters.action.action_filter import ActionFilter
+from rl_coach.spaces import BoxActionSpace
 
 
 class LinearBoxToBoxMap(ActionFilter):

rl_coach/filters/action/partial_discrete_action_space_map.py

@@ -16,10 +16,9 @@
 
 from typing import List
 
-from rl_coach.spaces import DiscreteActionSpace, ActionSpace
-
 from rl_coach.core_types import ActionType
 from rl_coach.filters.action.action_filter import ActionFilter
+from rl_coach.spaces import DiscreteActionSpace, ActionSpace
 
 
 class PartialDiscreteActionSpaceMap(ActionFilter):

rl_coach/filters/filter.py

@@ -19,8 +19,8 @@ from collections import OrderedDict
 from copy import deepcopy
 from typing import Dict, Union, List
 
-from rl_coach.spaces import ActionSpace, RewardSpace, ObservationSpace
 from rl_coach.core_types import EnvResponse, ActionInfo, Transition
+from rl_coach.spaces import ActionSpace, RewardSpace, ObservationSpace
 from rl_coach.utils import force_list
 
 

rl_coach/filters/observation/observation_clipping_filter.py

@@ -16,10 +16,10 @@
 
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class ObservationClippingFilter(ObservationFilter):

rl_coach/filters/observation/observation_crop_filter.py

@@ -16,10 +16,10 @@
 from typing import Union, Tuple
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class ObservationCropFilter(ObservationFilter):

rl_coach/filters/observation/observation_move_axis_filter.py

@@ -15,10 +15,10 @@
 #
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace, PlanarMapsObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace, PlanarMapsObservationSpace
 
 
 class ObservationMoveAxisFilter(ObservationFilter):

rl_coach/filters/observation/observation_normalization_filter.py

@@ -16,11 +16,11 @@
 from typing import List
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.architectures.tensorflow_components.shared_variables import SharedRunningStats
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class ObservationNormalizationFilter(ObservationFilter):

rl_coach/filters/observation/observation_reduction_by_sub_parts_name_filter.py

@@ -17,10 +17,9 @@ import copy
 from enum import Enum
 from typing import List
 
-from rl_coach.spaces import ObservationSpace, VectorObservationSpace
-
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace, VectorObservationSpace
 
 
 class ObservationReductionBySubPartsNameFilter(ObservationFilter):

rl_coach/filters/observation/observation_rescale_size_by_factor_filter.py

@@ -17,10 +17,10 @@
 from enum import Enum
 
 import scipy.ndimage
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 # imresize interpolation types as defined by scipy here:

rl_coach/filters/observation/observation_rescale_to_size_filter.py

@@ -19,10 +19,10 @@ from enum import Enum
 
 import numpy as np
 import scipy.ndimage
-from rl_coach.spaces import ObservationSpace, PlanarMapsObservationSpace, ImageObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace, PlanarMapsObservationSpace, ImageObservationSpace
 
 
 # imresize interpolation types as defined by scipy here:

rl_coach/filters/observation/observation_rgb_to_y_filter.py

@@ -14,10 +14,9 @@
 # limitations under the License.
 #
 
-from rl_coach.spaces import ObservationSpace
-
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class ObservationRGBToYFilter(ObservationFilter):

rl_coach/filters/observation/observation_squeeze_filter.py

@@ -15,10 +15,10 @@
 #
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class ObservationSqueezeFilter(ObservationFilter):

rl_coach/filters/observation/observation_stacking_filter.py

@@ -18,10 +18,10 @@ import copy
 from collections import deque
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class LazyStack(object):

rl_coach/filters/observation/observation_to_uint8_filter.py

@@ -15,10 +15,10 @@
 #
 
 import numpy as np
-from rl_coach.spaces import ObservationSpace
 
 from rl_coach.core_types import ObservationType
 from rl_coach.filters.observation.observation_filter import ObservationFilter
+from rl_coach.spaces import ObservationSpace
 
 
 class ObservationToUInt8Filter(ObservationFilter):

rl_coach/filters/reward/reward_clipping_filter.py

@@ -15,10 +15,10 @@
 #
 
 import numpy as np
-from rl_coach.spaces import RewardSpace
 
 from rl_coach.core_types import RewardType
 from rl_coach.filters.reward.reward_filter import RewardFilter
+from rl_coach.spaces import RewardSpace
 
 
 class RewardClippingFilter(RewardFilter):

rl_coach/filters/reward/reward_normalization_filter.py

@@ -16,11 +16,11 @@
 
 
 import numpy as np
-from rl_coach.spaces import RewardSpace
 
 from rl_coach.architectures.tensorflow_components.shared_variables import SharedRunningStats
 from rl_coach.core_types import RewardType
 from rl_coach.filters.reward.reward_filter import RewardFilter
+from rl_coach.spaces import RewardSpace
 
 
 class RewardNormalizationFilter(RewardFilter):

rl_coach/filters/reward/reward_rescale_filter.py

@@ -14,10 +14,9 @@
 # limitations under the License.
 #
 
-from rl_coach.spaces import RewardSpace
-
 from rl_coach.core_types import RewardType
 from rl_coach.filters.reward.reward_filter import RewardFilter
+from rl_coach.spaces import RewardSpace
 
 
 class RewardRescaleFilter(RewardFilter):