mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 22:08:16 +08:00
Move test folders under rllib/tune from test -> tests. (#4214)
This commit is contained in:
committed by
Philipp Moritz
parent
6f46edca51
commit
4b89eebfc7
@@ -23,7 +23,7 @@ import ray
|
||||
from ray import tune
|
||||
from ray.rllib.agents.ppo.ppo_policy_graph import PPOPolicyGraph
|
||||
from ray.rllib.models import Model, ModelCatalog
|
||||
from ray.rllib.test.test_multi_agent_env import MultiCartpole
|
||||
from ray.rllib.tests.test_multi_agent_env import MultiCartpole
|
||||
from ray.tune import run_experiments
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
|
||||
@@ -19,7 +19,7 @@ from ray.rllib.agents.dqn.dqn import DQNAgent
|
||||
from ray.rllib.agents.dqn.dqn_policy_graph import DQNPolicyGraph
|
||||
from ray.rllib.agents.ppo.ppo import PPOAgent
|
||||
from ray.rllib.agents.ppo.ppo_policy_graph import PPOPolicyGraph
|
||||
from ray.rllib.test.test_multi_agent_env import MultiCartpole
|
||||
from ray.rllib.tests.test_multi_agent_env import MultiCartpole
|
||||
from ray.tune.logger import pretty_print
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
|
||||
+1
-1
@@ -5,7 +5,7 @@ from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import ray
|
||||
from ray.rllib.test.test_multi_agent_env import make_multiagent
|
||||
from ray.rllib.tests.test_multi_agent_env import make_multiagent
|
||||
from ray.tune import run_experiments
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
+2
-2
@@ -13,8 +13,8 @@ from ray.rllib.agents.dqn import DQNAgent
|
||||
from ray.rllib.agents.pg import PGAgent
|
||||
from ray.rllib.evaluation.policy_evaluator import PolicyEvaluator
|
||||
from ray.rllib.env.external_env import ExternalEnv
|
||||
from ray.rllib.test.test_policy_evaluator import BadPolicyGraph, \
|
||||
MockPolicyGraph, MockEnv
|
||||
from ray.rllib.tests.test_policy_evaluator import (BadPolicyGraph,
|
||||
MockPolicyGraph, MockEnv)
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
|
||||
@@ -8,7 +8,7 @@ import numpy as np
|
||||
import ray
|
||||
from ray.rllib.utils.filter import RunningStat, MeanStdFilter
|
||||
from ray.rllib.utils import FilterManager
|
||||
from ray.rllib.test.mock_evaluator import _MockEvaluator
|
||||
from ray.rllib.tests.mock_evaluator import _MockEvaluator
|
||||
|
||||
|
||||
class RunningStatTest(unittest.TestCase):
|
||||
@@ -19,7 +19,7 @@ from ray.rllib.agents.pg.pg_policy_graph import PGPolicyGraph
|
||||
from ray.rllib.evaluation import SampleBatch
|
||||
from ray.rllib.offline import IOContext, JsonWriter, JsonReader
|
||||
from ray.rllib.offline.json_writer import _to_json
|
||||
from ray.rllib.test.test_multi_agent_env import MultiCartpole
|
||||
from ray.rllib.tests.test_multi_agent_env import MultiCartpole
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
SAMPLES = SampleBatch({
|
||||
+4
-4
@@ -10,10 +10,10 @@ import ray
|
||||
from ray.rllib.agents.pg import PGAgent
|
||||
from ray.rllib.agents.pg.pg_policy_graph import PGPolicyGraph
|
||||
from ray.rllib.agents.dqn.dqn_policy_graph import DQNPolicyGraph
|
||||
from ray.rllib.optimizers import SyncSamplesOptimizer, \
|
||||
SyncReplayOptimizer, AsyncGradientsOptimizer
|
||||
from ray.rllib.test.test_policy_evaluator import MockEnv, MockEnv2, \
|
||||
MockPolicyGraph
|
||||
from ray.rllib.optimizers import (SyncSamplesOptimizer, SyncReplayOptimizer,
|
||||
AsyncGradientsOptimizer)
|
||||
from ray.rllib.tests.test_policy_evaluator import (MockEnv, MockEnv2,
|
||||
MockPolicyGraph)
|
||||
from ray.rllib.evaluation.policy_evaluator import PolicyEvaluator
|
||||
from ray.rllib.evaluation.policy_graph import PolicyGraph
|
||||
from ray.rllib.evaluation.metrics import collect_metrics
|
||||
+1
-1
@@ -23,7 +23,7 @@ from ray.rllib.models.model import Model
|
||||
from ray.rllib.models.pytorch.fcnet import FullyConnectedNetwork
|
||||
from ray.rllib.models.pytorch.model import TorchModel
|
||||
from ray.rllib.rollout import rollout
|
||||
from ray.rllib.test.test_external_env import SimpleServing
|
||||
from ray.rllib.tests.test_external_env import SimpleServing
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
DICT_SPACE = spaces.Dict({
|
||||
+1
-1
@@ -14,7 +14,7 @@ from ray.rllib.agents.ppo.ppo_policy_graph import PPOPolicyGraph
|
||||
from ray.rllib.evaluation import SampleBatch
|
||||
from ray.rllib.evaluation.policy_evaluator import PolicyEvaluator
|
||||
from ray.rllib.optimizers import AsyncGradientsOptimizer, AsyncSamplesOptimizer
|
||||
from ray.rllib.test.mock_evaluator import _MockEvaluator
|
||||
from ray.rllib.tests.mock_evaluator import _MockEvaluator
|
||||
|
||||
|
||||
class AsyncOptimizerTest(unittest.TestCase):
|
||||
+2
-1
@@ -9,7 +9,8 @@ import sys
|
||||
|
||||
import ray
|
||||
from ray.rllib.agents.registry import get_agent_class
|
||||
from ray.rllib.test.test_multi_agent_env import MultiCartpole, MultiMountainCar
|
||||
from ray.rllib.tests.test_multi_agent_env import (MultiCartpole,
|
||||
MultiMountainCar)
|
||||
from ray.rllib.utils.error import UnsupportedSpaceException
|
||||
from ray.tune.registry import register_env
|
||||
|
||||
Reference in New Issue
Block a user