mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01:00
removing datasets + imports optimization
This commit is contained in:
@@ -15,11 +15,11 @@
|
||||
#
|
||||
from typing import Tuple, List
|
||||
|
||||
from rl_coach.base_parameters import AgentParameters, VisualizationParameters, TaskParameters, PresetValidationParameters
|
||||
from rl_coach.base_parameters import AgentParameters, VisualizationParameters, TaskParameters, \
|
||||
PresetValidationParameters
|
||||
from rl_coach.environments.environment import EnvironmentParameters, Environment
|
||||
from rl_coach.level_manager import LevelManager
|
||||
|
||||
from rl_coach.graph_managers.graph_manager import GraphManager, ScheduleParameters
|
||||
from rl_coach.level_manager import LevelManager
|
||||
from rl_coach.utils import short_dynamic_import
|
||||
|
||||
|
||||
|
||||
@@ -18,20 +18,19 @@ import copy
|
||||
import os
|
||||
import time
|
||||
from collections import OrderedDict
|
||||
from typing import List, Tuple
|
||||
from distutils.dir_util import copy_tree, remove_tree
|
||||
from typing import List, Tuple
|
||||
|
||||
import numpy as np
|
||||
from rl_coach.base_parameters import iterable_to_items, TaskParameters, DistributedTaskParameters, VisualizationParameters, \
|
||||
from rl_coach.base_parameters import iterable_to_items, TaskParameters, DistributedTaskParameters, \
|
||||
VisualizationParameters, \
|
||||
Parameters, PresetValidationParameters
|
||||
from rl_coach.core_types import TotalStepsCounter, RunPhase, PlayingStepsType, TrainingSteps, EnvironmentEpisodes, \
|
||||
EnvironmentSteps, \
|
||||
StepMethod
|
||||
from rl_coach.environments.environment import Environment
|
||||
from rl_coach.level_manager import LevelManager
|
||||
from rl_coach.utils import set_cpu
|
||||
|
||||
from rl_coach.logger import screen, Logger
|
||||
from rl_coach.utils import set_cpu
|
||||
|
||||
|
||||
class ScheduleParameters(Parameters):
|
||||
|
||||
@@ -13,17 +13,16 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
import numpy as np
|
||||
from typing import List, Union, Tuple
|
||||
|
||||
from rl_coach.base_parameters import AgentParameters, VisualizationParameters, TaskParameters, PresetValidationParameters
|
||||
from rl_coach.base_parameters import AgentParameters, VisualizationParameters, TaskParameters, \
|
||||
PresetValidationParameters
|
||||
from rl_coach.core_types import EnvironmentSteps
|
||||
from rl_coach.environments.environment import EnvironmentParameters, Environment
|
||||
from rl_coach.graph_managers.graph_manager import GraphManager, ScheduleParameters
|
||||
from rl_coach.level_manager import LevelManager
|
||||
from rl_coach.utils import short_dynamic_import
|
||||
|
||||
from rl_coach.core_types import EnvironmentSteps
|
||||
from rl_coach.graph_managers.graph_manager import GraphManager, ScheduleParameters
|
||||
|
||||
|
||||
class HACGraphManager(GraphManager):
|
||||
"""
|
||||
|
||||
@@ -16,14 +16,14 @@
|
||||
|
||||
from typing import List, Union, Tuple
|
||||
|
||||
from rl_coach.base_parameters import AgentParameters, VisualizationParameters, TaskParameters, PresetValidationParameters
|
||||
from rl_coach.base_parameters import AgentParameters, VisualizationParameters, TaskParameters, \
|
||||
PresetValidationParameters
|
||||
from rl_coach.core_types import EnvironmentSteps
|
||||
from rl_coach.environments.environment import EnvironmentParameters, Environment
|
||||
from rl_coach.graph_managers.graph_manager import GraphManager, ScheduleParameters
|
||||
from rl_coach.level_manager import LevelManager
|
||||
from rl_coach.utils import short_dynamic_import
|
||||
|
||||
from rl_coach.core_types import EnvironmentSteps
|
||||
from rl_coach.graph_managers.graph_manager import GraphManager, ScheduleParameters
|
||||
|
||||
|
||||
class HRLGraphManager(GraphManager):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user