mirror of
https://github.com/wassname/ray.git
synced 2026-09-10 12:38:43 +08:00
[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:
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user