removing datasets + imports optimization

rl_coach/architectures/tensorflow_components/embedders/embedder.py

@@ -18,10 +18,9 @@ from typing import List, Union
 
 import numpy as np
 import tensorflow as tf
-from rl_coach.architectures.tensorflow_components.shared_variables import SharedRunningStats
-from rl_coach.base_parameters import EmbedderScheme
 
 from rl_coach.architectures.tensorflow_components.architecture import batchnorm_activation_dropout
+from rl_coach.base_parameters import EmbedderScheme
 from rl_coach.core_types import InputEmbedding
 
 

rl_coach/architectures/tensorflow_components/embedders/image_embedder.py

@@ -17,10 +17,10 @@
 from typing import List
 
 import tensorflow as tf
-from rl_coach.architectures.tensorflow_components.architecture import Conv2d
-from rl_coach.base_parameters import EmbedderScheme
 
+from rl_coach.architectures.tensorflow_components.architecture import Conv2d
 from rl_coach.architectures.tensorflow_components.embedders.embedder import InputEmbedder
+from rl_coach.base_parameters import EmbedderScheme
 from rl_coach.core_types import InputImageEmbedding
 
 

rl_coach/architectures/tensorflow_components/embedders/vector_embedder.py

@@ -17,10 +17,10 @@
 from typing import List
 
 import tensorflow as tf
-from rl_coach.architectures.tensorflow_components.architecture import Dense
-from rl_coach.base_parameters import EmbedderScheme
 
+from rl_coach.architectures.tensorflow_components.architecture import Dense
 from rl_coach.architectures.tensorflow_components.embedders.embedder import InputEmbedder
+from rl_coach.base_parameters import EmbedderScheme
 from rl_coach.core_types import InputVectorEmbedding