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.train.examples.torch_fashion_mnist_example import train_func
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):

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

@ -1,6 +1,6 @@
from ray.air import ScalingConfig, RunConfig, session
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