hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-04 17:41:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[AIR] Fix deprecated import of MLflowLoggerCallback (#28247)

Browse source 
Signed-off-by: Kilian Lieret <kilian.lieret@posteo.de>
This commit is contained in:
Kilian Lieret 2022-09-01 20:55:59 -04:00 • committed by GitHub
parent b83f10dbde
commit 77722b86fd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/ray/train/examples/mlflow_fashion_mnist_example.py
View file

@ -3,7 +3,7 @@ import argparse
from ray.air.config import RunConfig, ScalingConfig from ray.air.config import RunConfig, ScalingConfig
from ray.train.examples.torch_fashion_mnist_example import train_func from ray.train.examples.torch_fashion_mnist_example import train_func
from ray.train.torch import TorchTrainer from ray.train.torch import TorchTrainer
from ray.tune.integration.mlflow import MLflowLoggerCallback from ray.air.callbacks.mlflow import MLflowLoggerCallback
def main(num_workers=2, use_gpu=False): def main(num_workers=2, use_gpu=False):

2
python/ray/train/examples/mlflow_simple_example.py
View file

@ -1,6 +1,6 @@
from ray.air import ScalingConfig, RunConfig, session from ray.air import ScalingConfig, RunConfig, session
from ray.train.torch import TorchTrainer from ray.train.torch import TorchTrainer
from ray.tune.integration.mlflow import MLflowLoggerCallback from ray.air.callbacks.mlflow import MLflowLoggerCallback
from ray.tune.logger import TBXLoggerCallback from ray.tune.logger import TBXLoggerCallback