[tune] logger refactor part 1: move classes and utilities to own files (#11746)

* [tune] logger refactor part 1: move classes and utilities to own files

* Fix circular dependency

* Remove uneeded pretty print copy

* Apply suggestions from code review
This commit is contained in:
Kai Fricke
2020-11-03 07:48:09 -08:00
committed by GitHub
parent 5af745c90d
commit f7b19c41e3
20 changed files with 481 additions and 375 deletions
+1 -1
View File
@@ -19,7 +19,7 @@ from ray.tune.ray_trial_executor import RayTrialExecutor
from ray.tune.resources import Resources
from ray.tune.suggest import BasicVariantGenerator
from ray.tune.syncer import CloudSyncer
from ray.tune.trainable import TrainableUtil
from ray.tune.utils.trainable import TrainableUtil
from ray.tune.trial import Trial
from ray.tune.trial_runner import TrialRunner
from ray.tune.utils.mock import (MockDurableTrainer, MockRemoteTrainer,
+1 -1
View File
@@ -9,7 +9,7 @@ from ray.rllib import _register_all
from ray import tune
from ray.tune.logger import NoopLogger
from ray.tune.trainable import TrainableUtil
from ray.tune.utils.trainable import TrainableUtil
from ray.tune.function_runner import with_parameters, wrap_function, \
FuncCheckpointUtil
from ray.tune.result import TRAINING_ITERATION
+1 -1
View File
@@ -5,7 +5,7 @@ import unittest
import ray.utils
from ray.tune.trainable import TrainableUtil
from ray.tune.utils.trainable import TrainableUtil
class TrainableUtilTest(unittest.TestCase):
@@ -13,7 +13,8 @@ from ray.tune.ray_trial_executor import RayTrialExecutor
from ray.tune.result import TRAINING_ITERATION
from ray.tune.trial import Trial
from ray.tune.trial_runner import Callback, TrialRunner
from ray.tune.callback import Callback
from ray.tune.trial_runner import TrialRunner
class TestCallback(Callback):
@@ -45,7 +46,7 @@ class TestCallback(Callback):
def on_trial_complete(self, **info):
self.state["trial_complete"] = info
def on_trial_fail(self, **info):
def on_trial_error(self, **info):
self.state["trial_fail"] = info