move evaluator

python/ray/rllib/dqn/base_evaluator.py

@@ -9,8 +9,7 @@
 from ray.rllib.dqn import models
 from ray.rllib.dqn.common.wrappers import wrap_dqn
 from ray.rllib.dqn.common.schedules import LinearSchedule
-from ray.rllib.evaluator import TFMultiGPUSupport
-from ray.rllib.optimizers import SampleBatch
+from ray.rllib.optimizers import SampleBatch, TFMultiGPUSupport
 
 
 class DQNEvaluator(TFMultiGPUSupport):

python/ray/rllib/optimizers/__init__.py

@@ -2,8 +2,9 @@
 from ray.rllib.optimizers.local_sync import LocalSyncOptimizer
 from ray.rllib.optimizers.multi_gpu import LocalMultiGPUOptimizer
 from ray.rllib.optimizers.sample_batch import SampleBatch
+from ray.rllib.optimizers.evaluator import Evaluator, TFMultiGPUSupport
 
 
 __all__ = [
     "AsyncOptimizer", "LocalSyncOptimizer", "LocalMultiGPUOptimizer",
-    "SampleBatch"]
+    "SampleBatch", "Evaluator", "TFMultiGPUSupport"]

python/ray/rllib/optimizers/multi_gpu.py

@@ -7,7 +7,7 @@
 import tensorflow as tf
 
 import ray
-from ray.rllib.evaluator import TFMultiGPUSupport
+from ray.rllib.optimizers.evaluator import TFMultiGPUSupport
 from ray.rllib.optimizers.optimizer import Optimizer
 from ray.rllib.optimizers.sample_batch import SampleBatch
 from ray.rllib.parallel import LocalSyncParallelOptimizer